langchain updates

src/opengradient/agents/__init__.py

@@ -6,15 +6,24 @@
 into existing applications and agent frameworks.
 """
 
+from typing import Optional, Union
+
+from ..client.llm import LLM
 from ..types import TEE_LLM, x402SettlementMode
 from .og_langchain import *
 
 
 def langchain_adapter(
-    private_key: str,
-    model_cid: TEE_LLM,
+    private_key: Optional[str] = None,
+    model_cid: Optional[Union[TEE_LLM, str]] = None,
+    model: Optional[Union[TEE_LLM, str]] = None,
     max_tokens: int = 300,
+    temperature: float = 0.0,
     x402_settlement_mode: x402SettlementMode = x402SettlementMode.BATCH_HASHED,
+    client: Optional[LLM] = None,
+    rpc_url: Optional[str] = None,
+    tee_registry_address: Optional[str] = None,
+    llm_server_url: Optional[str] = None,
 ) -> OpenGradientChatModel:
     """
     Returns an OpenGradient LLM that implements LangChain's LLM interface
@@ -23,8 +32,14 @@ def langchain_adapter(
     return OpenGradientChatModel(
         private_key=private_key,
         model_cid=model_cid,
+        model=model,
         max_tokens=max_tokens,
+        temperature=temperature,
         x402_settlement_mode=x402_settlement_mode,
+        client=client,
+        rpc_url=rpc_url,
+        tee_registry_address=tee_registry_address,
+        llm_server_url=llm_server_url,
     )