Merge branch 'master' into bagatur/update_to_3_15

langchain-ai · Oct 31, 2024 · 6320c85 · 6320c85
2 parents 888961f + f94125a
commit 6320c85
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/libs/community/langchain_community/adapters/openai.py b/libs/community/langchain_community/adapters/openai.py
@@ -180,6 +180,12 @@ def _convert_message_chunk(chunk: BaseMessageChunk, i: int) -> dict:
             # not missing, but None.
             if i == 0:
                 _dict["content"] = None
+        if "tool_calls" in chunk.additional_kwargs:
+            _dict["tool_calls"] = chunk.additional_kwargs["tool_calls"]
+            # If the first chunk is tool calls, the content is not empty string,
+            # not missing, but None.
+            if i == 0:
+                _dict["content"] = None
         else:
             _dict["content"] = chunk.content
     else:

diff --git a/libs/community/langchain_community/utilities/polygon.py b/libs/community/langchain_community/utilities/polygon.py
@@ -45,7 +45,7 @@ def get_financials(self, ticker: str) -> Optional[dict]:
         data = response.json()
 
         status = data.get("status", None)
-        if status not in ("OK", "STOCKBUSINESS"):
+        if status not in ("OK", "STOCKBUSINESS", "STOCKSBUSINESS"):
             raise ValueError(f"API Error: {data}")
 
         return data.get("results", None)
@@ -61,7 +61,7 @@ def get_last_quote(self, ticker: str) -> Optional[dict]:
         data = response.json()
 
         status = data.get("status", None)
-        if status not in ("OK", "STOCKBUSINESS"):
+        if status not in ("OK", "STOCKBUSINESS", "STOCKSBUSINESS"):
             raise ValueError(f"API Error: {data}")
 
         return data.get("results", None)
@@ -82,7 +82,7 @@ def get_ticker_news(self, ticker: str) -> Optional[dict]:
         data = response.json()
 
         status = data.get("status", None)
-        if status not in ("OK", "STOCKBUSINESS"):
+        if status not in ("OK", "STOCKBUSINESS", "STOCKSBUSINESS"):
             raise ValueError(f"API Error: {data}")
 
         return data.get("results", None)
@@ -116,7 +116,7 @@ def get_aggregates(self, ticker: str, **kwargs: Any) -> Optional[dict]:
         data = response.json()
 
         status = data.get("status", None)
-        if status not in ("OK", "STOCKBUSINESS"):
+        if status not in ("OK", "STOCKBUSINESS", "STOCKSBUSINESS"):
             raise ValueError(f"API Error: {data}")
 
         return data.get("results", None)