langchain-fireworks 0.2.0.dev0__tar.gz → 0.2.0.dev2__tar.gz

{langchain_fireworks-0.2.0.dev0 → langchain_fireworks-0.2.0.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langchain-fireworks
-Version: 0.2.0.dev0
+Version: 0.2.0.dev2
 Summary: An integration package connecting Fireworks and LangChain
 Home-page: https://github.com/langchain-ai/langchain
 License: MIT
@@ -13,7 +13,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: aiohttp (>=3.9.1,<4.0.0)
 Requires-Dist: fireworks-ai (>=0.13.0)
-Requires-Dist: langchain-core (>=0.3.0.dev1,<0.4.0)
+Requires-Dist: langchain-core (>=0.3.0.dev5,<0.4.0)
 Requires-Dist: openai (>=1.10.0,<2.0.0)
 Requires-Dist: requests (>=2,<3)
 Project-URL: Repository, https://github.com/langchain-ai/langchain

{langchain_fireworks-0.2.0.dev0 → langchain_fireworks-0.2.0.dev2}/langchain_fireworks/chat_models.py

@@ -464,7 +464,7 @@ class ChatFireworks(BaseChatModel):
         default_chunk_class: Type[BaseMessageChunk] = AIMessageChunk
         for chunk in self.client.create(messages=message_dicts, **params):
             if not isinstance(chunk, dict):
-                chunk = chunk.dict()
+                chunk = chunk.model_dump()
             if len(chunk["choices"]) == 0:
                 continue
             choice = chunk["choices"][0]
@@ -520,7 +520,7 @@ class ChatFireworks(BaseChatModel):
     def _create_chat_result(self, response: Union[dict, BaseModel]) -> ChatResult:
         generations = []
         if not isinstance(response, dict):
-            response = response.dict()
+            response = response.model_dump()
         token_usage = response.get("usage", {})
         for res in response["choices"]:
             message = _convert_dict_to_message(res["message"])
@@ -558,7 +558,7 @@ class ChatFireworks(BaseChatModel):
         default_chunk_class: Type[BaseMessageChunk] = AIMessageChunk
         async for chunk in self.async_client.acreate(messages=message_dicts, **params):
             if not isinstance(chunk, dict):
-                chunk = chunk.dict()
+                chunk = chunk.model_dump()
             if len(chunk["choices"]) == 0:
                 continue
             choice = chunk["choices"][0]

{langchain_fireworks-0.2.0.dev0 → langchain_fireworks-0.2.0.dev2}/pyproject.toml

@@ -1,10 +1,10 @@
 [build-system]
-requires = [ "poetry-core>=1.0.0",]
+requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "langchain-fireworks"
-version = "0.2.0.dev0"
+version = "0.2.0.dev2"
 description = "An integration package connecting Fireworks and LangChain"
 authors = []
 readme = "README.md"
@@ -20,21 +20,25 @@ disallow_untyped_defs = "True"
 
 [tool.poetry.dependencies]
 python = ">=3.9,<4.0"
-langchain-core = { version = "^0.3.0.dev1", allow-prereleases = true }
+langchain-core = { version = "^0.3.0.dev5", allow-prereleases = true }
 fireworks-ai = ">=0.13.0"
 openai = "^1.10.0"
 requests = "^2"
 aiohttp = "^3.9.1"
 
 [tool.ruff.lint]
-select = [ "E", "F", "I", "T201",]
+select = ["E", "F", "I", "T201"]
 
 [tool.coverage.run]
-omit = [ "tests/*",]
+omit = ["tests/*"]
 
 [tool.pytest.ini_options]
 addopts = "--snapshot-warn-unused --strict-markers --strict-config --durations=5"
-markers = [ "requires: mark tests as requiring a specific library", "asyncio: mark tests as requiring asyncio", "compile: mark placeholder test used to compile integration tests without running them",]
+markers = [
+    "requires: mark tests as requiring a specific library",
+    "asyncio: mark tests as requiring asyncio",
+    "compile: mark placeholder test used to compile integration tests without running them",
+]
 asyncio_mode = "auto"
 
 [tool.poetry.group.test]