opengradient 0.3.9__tar.gz → 0.3.11__tar.gz

{opengradient-0.3.9/src/opengradient.egg-info → opengradient-0.3.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: opengradient
-Version: 0.3.9
+Version: 0.3.11
 Summary: Python SDK for OpenGradient decentralized model management & inference services
 Author-email: OpenGradient <oliver@opengradient.ai>
 License: MIT License
@@ -87,7 +87,6 @@ Requires-Dist: keyring==24.3.1
 Requires-Dist: more-itertools==10.5.0
 Requires-Dist: msgpack==1.1.0
 Requires-Dist: multidict==6.1.0
-Requires-Dist: numpy==2.1.1
 Requires-Dist: packaging==24.1
 Requires-Dist: pandas==2.2.3
 Requires-Dist: parsimonious==0.10.0

{opengradient-0.3.9 → opengradient-0.3.11}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "opengradient"
-version = "0.3.9"
+version = "0.3.11"
 description = "Python SDK for OpenGradient decentralized model management & inference services"
 authors = [{name = "OpenGradient", email = "oliver@opengradient.ai"}]
 license = {file = "LICENSE"}
@@ -71,7 +71,6 @@ dependencies = [
     "more-itertools==10.5.0",
     "msgpack==1.1.0",
     "multidict==6.1.0",
-    "numpy==2.1.1",
     "packaging==24.1",
     "pandas==2.2.3",
     "parsimonious==0.10.0",

{opengradient-0.3.9 → opengradient-0.3.11}/src/opengradient/__init__.py

@@ -3,8 +3,9 @@ from typing import Dict, List, Optional, Tuple
 from .client import Client
 from .defaults import DEFAULT_INFERENCE_CONTRACT_ADDRESS, DEFAULT_RPC_URL
 from .types import InferenceMode, LLM
+from . import llm
 
-__version__ = "0.3.9"
+__version__ = "0.3.11"
 
 _client = None
 

opengradient-0.3.11/src/opengradient/llm/__init__.py

@@ -0,0 +1,5 @@
+from .chat import *
+
+__all__ = [
+    'OpenGradientChatModel'
+]

opengradient-0.3.11/src/opengradient/llm/chat.py

@@ -0,0 +1,119 @@
+from typing import List, Dict, Optional, Any, Sequence, Union
+import json
+
+from langchain.chat_models.base import BaseChatModel
+from langchain.schema import (
+    AIMessage,
+    HumanMessage,
+    SystemMessage,
+    BaseMessage,
+    ChatResult,
+    ChatGeneration,
+)
+from langchain_core.callbacks.manager import CallbackManagerForLLMRun
+from langchain_core.tools import BaseTool
+from langchain_core.messages import ToolCall
+
+from opengradient import Client
+from opengradient.defaults import DEFAULT_RPC_URL, DEFAULT_INFERENCE_CONTRACT_ADDRESS
+
+
+class OpenGradientChatModel(BaseChatModel):
+    """OpenGradient adapter class for LangChain chat model"""
+
+    client: Client = None
+    model_cid: str = None
+    max_tokens: int = None
+    tools: List[Dict] = []
+
+    def __init__(self, private_key: str, model_cid: str, max_tokens: int = 300):
+        super().__init__()
+        self.client = Client(
+            private_key=private_key,
+            rpc_url=DEFAULT_RPC_URL,
+            contract_address=DEFAULT_INFERENCE_CONTRACT_ADDRESS,
+            email=None,
+            password=None)
+        self.model_cid = model_cid
+        self.max_tokens = max_tokens
+
+    @property
+    def _llm_type(self) -> str:
+        return "opengradient"
+    
+    def bind_tools(
+        self,
+        tools: Sequence[Union[BaseTool, Dict]],
+    ) -> "OpenGradientChatModel":
+        """Bind tools to the model."""
+        tool_dicts = []
+        for tool in tools:
+            if isinstance(tool, BaseTool):
+                tool_dicts.append({
+                    "type": "function",
+                    "function": {
+                        "name": tool.name,
+                        "description": tool.description,
+                        "parameters": tool.args_schema.schema() if hasattr(tool, "args_schema") else {}
+                    }
+                })
+            else:
+                tool_dicts.append(tool)
+        
+        self.tools = tool_dicts
+        return self
+
+    def _generate(
+        self,
+        messages: List[BaseMessage],
+        stop: Optional[List[str]] = None,
+        run_manager: Optional[CallbackManagerForLLMRun] = None,
+        **kwargs: Any,
+    ) -> ChatResult:
+        chat_messages = []
+        for message in messages:
+            if isinstance(message, SystemMessage):
+                chat_messages.append({"role": "system", "content": message.content})
+            elif isinstance(message, HumanMessage):
+                chat_messages.append({"role": "user", "content": message.content})
+            elif isinstance(message, AIMessage):
+                chat_messages.append({"role": "assistant", "content": message.content})
+
+        tx_hash, finish_reason, chat_response = self.client.llm_chat(
+            model_cid=self.model_cid,
+            messages=chat_messages,
+            stop_sequence=stop,
+            max_tokens=self.max_tokens,
+            tools=self.tools
+        )
+
+        if "tool_calls" in chat_response:
+            tool_calls = []
+            for tool_call in chat_response["tool_calls"]:
+                tool_calls.append(
+                    ToolCall(
+                        id=tool_call.get("id", ""),
+                        name=tool_call["name"],
+                        args=json.loads(tool_call["arguments"])
+                    )
+                )
+            
+            message = AIMessage(
+                content='',
+                tool_calls=tool_calls
+            )
+        else:
+            message = AIMessage(content=chat_response["content"])
+        
+        return ChatResult(
+            generations=[ChatGeneration(
+                message=message,
+                generation_info={"finish_reason": finish_reason}
+            )]
+        )
+
+    @property
+    def _identifying_params(self) -> Dict[str, Any]:
+        return {
+            "model_name": self.model_cid,
+        }

{opengradient-0.3.9 → opengradient-0.3.11/src/opengradient.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: opengradient
-Version: 0.3.9
+Version: 0.3.11
 Summary: Python SDK for OpenGradient decentralized model management & inference services
 Author-email: OpenGradient <oliver@opengradient.ai>
 License: MIT License
@@ -87,7 +87,6 @@ Requires-Dist: keyring==24.3.1
 Requires-Dist: more-itertools==10.5.0
 Requires-Dist: msgpack==1.1.0
 Requires-Dist: multidict==6.1.0
-Requires-Dist: numpy==2.1.1
 Requires-Dist: packaging==24.1
 Requires-Dist: pandas==2.2.3
 Requires-Dist: parsimonious==0.10.0

{opengradient-0.3.9 → opengradient-0.3.11}/src/opengradient.egg-info/SOURCES.txt

@@ -15,4 +15,6 @@ src/opengradient.egg-info/dependency_links.txt
 src/opengradient.egg-info/entry_points.txt
 src/opengradient.egg-info/requires.txt
 src/opengradient.egg-info/top_level.txt
-src/opengradient/abi/inference.abi
+src/opengradient/abi/inference.abi
+src/opengradient/llm/__init__.py
+src/opengradient/llm/chat.py

{opengradient-0.3.9 → opengradient-0.3.11}/src/opengradient.egg-info/requires.txt

@@ -50,7 +50,6 @@ keyring==24.3.1
 more-itertools==10.5.0
 msgpack==1.1.0
 multidict==6.1.0
-numpy==2.1.1
 packaging==24.1
 pandas==2.2.3
 parsimonious==0.10.0