github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-22 06:39:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[minor]: Updating payload for pebblo discover API (#22309)

Browse Source 
**Description:** Updating response for pebblo discover API. Also
updating filed name case type
**Documentation:** N/A
**Unit tests:** N/A
This commit is contained in:
Dristy Srivastava 2024-06-04 04:06:17 +05:30 committed by GitHub
parent cbd5720011
commit ef3df45d9d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
libs/community/langchain_community/document_loaders/pebblo.py
View File

@ -327,17 +327,16 @@ class PebbloSafeLoader(BaseLoader):
try: try:
headers.update({"x-api-key": self.api_key}) headers.update({"x-api-key": self.api_key})
if pebblo_resp: if pebblo_resp:
pebblo_resp_docs = json.loads(pebblo_resp.text).get("ai_apps_data") pebblo_server_version = json.loads(pebblo_resp.text).get(
"pebblo_server_version"
)
payload.update( payload.update(
{ {
"pebblo_server_version": pebblo_resp_docs.get( "pebblo_server_version": pebblo_server_version,
"pebbloServerVersion" "pebblo_client_version": payload["plugin_version"],
),
"pebblo_client_version": pebblo_resp_docs.get(
"pebbloClientVersion"
),
} }
) )
payload.pop("plugin_version")
pebblo_cloud_url = f"{PEBBLO_CLOUD_URL}{APP_DISCOVER_URL}" pebblo_cloud_url = f"{PEBBLO_CLOUD_URL}{APP_DISCOVER_URL}"
pebblo_cloud_response = requests.post( pebblo_cloud_response = requests.post(
pebblo_cloud_url, headers=headers, json=payload, timeout=20 pebblo_cloud_url, headers=headers, json=payload, timeout=20