opengradient 0.3.13__tar.gz → 0.3.14__tar.gz

{opengradient-0.3.13 → opengradient-0.3.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: opengradient
-Version: 0.3.13
+Version: 0.3.14
 Summary: Python SDK for OpenGradient decentralized model management & inference services
 Project-URL: Homepage, https://opengradient.ai
 Author-email: OpenGradient <oliver@opengradient.ai>

{opengradient-0.3.13 → opengradient-0.3.14}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "opengradient"
-version = "0.3.13"
+version = "0.3.14"
 description = "Python SDK for OpenGradient decentralized model management & inference services"
 authors = [{name = "OpenGradient", email = "oliver@opengradient.ai"}]
 license = {file = "LICENSE"}

{opengradient-0.3.13 → opengradient-0.3.14}/src/opengradient/__init__.py

@@ -5,7 +5,7 @@ from .defaults import DEFAULT_INFERENCE_CONTRACT_ADDRESS, DEFAULT_RPC_URL
 from .types import InferenceMode, LLM
 from . import llm
 
-__version__ = "0.3.13"
+__version__ = "0.3.14"
 
 _client = None
 

{opengradient-0.3.13 → opengradient-0.3.14}/src/opengradient/llm/chat.py

@@ -10,6 +10,9 @@ from langchain.schema import (
     ChatResult,
     ChatGeneration,
 )
+from langchain_core.messages.tool import (
+    ToolMessage
+)
 from langchain_core.callbacks.manager import CallbackManagerForLLMRun
 from langchain_core.tools import BaseTool
 from langchain_core.messages import ToolCall
@@ -70,24 +73,36 @@ class OpenGradientChatModel(BaseChatModel):
         run_manager: Optional[CallbackManagerForLLMRun] = None,
         **kwargs: Any,
     ) -> ChatResult:
-        chat_messages = []
+
+        sdk_messages = []
         for message in messages:
             if isinstance(message, SystemMessage):
-                chat_messages.append({"role": "system", "content": message.content})
+                sdk_messages.append({"role": "system", "content": message.content})
             elif isinstance(message, HumanMessage):
-                chat_messages.append({"role": "user", "content": message.content})
+                sdk_messages.append({"role": "user", "content": message.content})
             elif isinstance(message, AIMessage):
-                chat_messages.append({"role": "assistant", "content": message.content})
+                sdk_messages.append({
+                    "role": "assistant", 
+                    "content": message.content, 
+                    "tool_calls": [{
+                        "id": call["id"],
+                        "name": call["name"],
+                        "arguments": json.dumps(call["args"])
+                    } for call in message.tool_calls]})
+            elif isinstance(message, ToolMessage):
+                sdk_messages.append({"role": "tool", "content": message.content, "tool_call_id": message.tool_call_id})
+            else:
+                raise ValueError(f"Unexpected message type: {message}")
 
-        tx_hash, finish_reason, chat_response = self.client.llm_chat(
+        _, finish_reason, chat_response = self.client.llm_chat(
             model_cid=self.model_cid,
-            messages=chat_messages,
+            messages=sdk_messages,
             stop_sequence=stop,
             max_tokens=self.max_tokens,
             tools=self.tools
         )
 
-        if "tool_calls" in chat_response:
+        if "tool_calls" in chat_response and chat_response["tool_calls"]:
             tool_calls = []
             for tool_call in chat_response["tool_calls"]:
                 tool_calls.append(
@@ -104,7 +119,7 @@ class OpenGradientChatModel(BaseChatModel):
             )
         else:
             message = AIMessage(content=chat_response["content"])
-        
+
         return ChatResult(
             generations=[ChatGeneration(
                 message=message,