langchain-xache 0.1.1__py3-none-any.whl → 0.2.1__py3-none-any.whl

{langchain_xache-0.1.1.dist-info → langchain_xache-0.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-xache
-Version: 0.1.1
+Version: 0.2.1
 Summary: LangChain integration for Xache Protocol - verifiable AI agent memory
 Author-email: Xache Protocol <dev@xache.xyz>
 License: MIT
@@ -190,9 +190,8 @@ Xache supports [ERC-8004](https://eips.ethereum.org/EIPS/eip-8004) for portable,
 ## Resources
 
 - [Documentation](https://docs.xache.xyz)
-- [API Reference](https://docs.xache.xyz/api)
-- [GitHub](https://github.com/oliveskin/xache)
-- [Discord](https://discord.gg/xache)
+- [GitHub](https://github.com/xacheai/xache-protocol)
+- [Website](https://xache.xyz)
 
 ## License
 

{langchain_xache-0.1.1.dist-info → langchain_xache-0.2.1.dist-info}/RECORD

@@ -2,11 +2,11 @@ xache_langchain/__init__.py,sha256=OYzpiCnA75ToCTVmQcqut7kGygD9lQhjrUTiFZU5S88,1
 xache_langchain/_async_utils.py,sha256=Srhe4rJ1S4W6E0sDfgsTqMdfMFr0Je9scJ3cFNa8fWA,1699
 xache_langchain/chat_history.py,sha256=MS6w8fyowWlLvvtZ0RBQWNygBytM3cnasZ6C0KwL4sE,6730
 xache_langchain/collective.py,sha256=OyZKvfOx20ETK236P36fqAZ6ahLyos6FYs81IOKjti0,8092
-xache_langchain/extraction.py,sha256=tFJtL3Sk1XK7UUSmlLw3eMsPA9Lh5kI-G8EhQ1Uiuf0,7407
+xache_langchain/extraction.py,sha256=4mrugy9zAsoO1SV_5gOBq2FF7A9D8U67x1Rek8drmWg,9734
 xache_langchain/memory.py,sha256=71hxXcUtsyaAo67_xBmjel43EvhgDiCENcD6AvJxMBo,6196
 xache_langchain/reputation.py,sha256=b1GVceH1QPqB7rjvTj295QllxiJ5W0rQ5rUoKJ5Np7A,7389
 xache_langchain/retriever.py,sha256=qw7CSP7zR6g7mn9DUYHbId9S9K564kKXqKR1WnnHkP0,7607
-langchain_xache-0.1.1.dist-info/METADATA,sha256=9KWfldb-IxnLgHgf3GWTkLGGkm-P4IVUawXt5HBbU-Q,5308
-langchain_xache-0.1.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-langchain_xache-0.1.1.dist-info/top_level.txt,sha256=in69PSq9agqGIAyShkm5ZVg9n0ks76QlD1tGaws9efA,16
-langchain_xache-0.1.1.dist-info/RECORD,,
+langchain_xache-0.2.1.dist-info/METADATA,sha256=dW4FUcf5_-BIRSBxt7dJ1HFgSf0e2kqaNNfJ71oJDqY,5261
+langchain_xache-0.2.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+langchain_xache-0.2.1.dist-info/top_level.txt,sha256=in69PSq9agqGIAyShkm5ZVg9n0ks76QlD1tGaws9efA,16
+langchain_xache-0.2.1.dist-info/RECORD,,

xache_langchain/extraction.py

@@ -58,6 +58,15 @@ class XacheExtractor:
         ```
     """
 
+    # Supported LLM providers for api-key mode
+    SUPPORTED_PROVIDERS = [
+        "anthropic", "openai", "google", "mistral", "groq",
+        "together", "fireworks", "cohere", "xai", "deepseek"
+    ]
+
+    # Supported API formats for endpoint mode
+    SUPPORTED_FORMATS = ["openai", "anthropic", "cohere"]
+
     def __init__(
         self,
         wallet_address: str,
@@ -66,8 +75,11 @@ class XacheExtractor:
         chain: str = "base",
         mode: str = "xache-managed",
         llm_api_key: Optional[str] = None,
+        llm_provider: str = "anthropic",
         llm_endpoint: Optional[str] = None,
-        llm_model: str = "claude-3-haiku-20240307",
+        llm_endpoint_format: str = "openai",
+        llm_auth_token: Optional[str] = None,
+        llm_model: Optional[str] = None,
     ):
         """
         Initialize Xache extractor.
@@ -77,16 +89,28 @@ class XacheExtractor:
             private_key: Private key for signing
             api_url: Xache API URL (defaults to XACHE_API_URL env var or https://api.xache.xyz)
             chain: Blockchain (base, solana)
-            mode: Extraction mode (xache-managed, api-key, endpoint)
+            mode: Extraction mode:
+                - 'xache-managed': Xache provides LLM ($0.011)
+                - 'api-key': Use major provider with your API key ($0.002)
+                - 'endpoint': Use custom endpoint ($0.002)
             llm_api_key: Your LLM API key (required for api-key mode)
-            llm_endpoint: Your LLM endpoint (required for endpoint mode)
-            llm_model: LLM model to use
+            llm_provider: Provider for api-key mode. Supported:
+                anthropic, openai, google, mistral, groq, together,
+                fireworks, cohere, xai, deepseek
+            llm_endpoint: Custom endpoint URL (required for endpoint mode)
+            llm_endpoint_format: API format for endpoint mode (openai, anthropic, cohere)
+            llm_auth_token: Auth token for endpoint mode
+            llm_model: Model to use (optional, uses provider default)
         """
         # Validate mode-specific requirements
         if mode == "api-key" and not llm_api_key:
             raise ValueError("llm_api_key is required when mode is 'api-key'")
+        if mode == "api-key" and llm_provider not in self.SUPPORTED_PROVIDERS:
+            raise ValueError(f"llm_provider must be one of: {', '.join(self.SUPPORTED_PROVIDERS)}")
         if mode == "endpoint" and not llm_endpoint:
             raise ValueError("llm_endpoint is required when mode is 'endpoint'")
+        if mode == "endpoint" and llm_endpoint_format not in self.SUPPORTED_FORMATS:
+            raise ValueError(f"llm_endpoint_format must be one of: {', '.join(self.SUPPORTED_FORMATS)}")
 
         self.wallet_address = wallet_address
         self.private_key = private_key
@@ -94,7 +118,10 @@ class XacheExtractor:
         self.chain = chain
         self.mode = mode
         self.llm_api_key = llm_api_key
+        self.llm_provider = llm_provider
         self.llm_endpoint = llm_endpoint
+        self.llm_endpoint_format = llm_endpoint_format
+        self.llm_auth_token = llm_auth_token
         self.llm_model = llm_model
 
         # Build DID
@@ -131,20 +158,28 @@ class XacheExtractor:
                 if self.mode == "api-key":
                     llm_config = {
                         "type": "api-key",
+                        "provider": self.llm_provider,
                         "apiKey": self.llm_api_key,
-                        "model": self.llm_model,
                     }
+                    if self.llm_model:
+                        llm_config["model"] = self.llm_model
                 elif self.mode == "endpoint":
                     llm_config = {
                         "type": "endpoint",
-                        "endpoint": self.llm_endpoint,
-                        "model": self.llm_model,
+                        "url": self.llm_endpoint,
+                        "format": self.llm_endpoint_format,
                     }
+                    if self.llm_auth_token:
+                        llm_config["authToken"] = self.llm_auth_token
+                    if self.llm_model:
+                        llm_config["model"] = self.llm_model
                 else:
                     llm_config = {
                         "type": "xache-managed",
-                        "model": self.llm_model,
+                        "provider": "anthropic",
                     }
+                    if self.llm_model:
+                        llm_config["model"] = self.llm_model
 
                 result = await client.extraction.extract(
                     trace=trace,
@@ -171,23 +206,32 @@ class XacheExtractor:
     ) -> ExtractionResult:
         """Async extract memories from text"""
         async with self._client as client:
+            # Build LLM config based on mode
             if self.mode == "api-key":
                 llm_config = {
                     "type": "api-key",
+                    "provider": self.llm_provider,
                     "apiKey": self.llm_api_key,
-                    "model": self.llm_model,
                 }
+                if self.llm_model:
+                    llm_config["model"] = self.llm_model
             elif self.mode == "endpoint":
                 llm_config = {
                     "type": "endpoint",
-                    "endpoint": self.llm_endpoint,
-                    "model": self.llm_model,
+                    "url": self.llm_endpoint,
+                    "format": self.llm_endpoint_format,
                 }
+                if self.llm_auth_token:
+                    llm_config["authToken"] = self.llm_auth_token
+                if self.llm_model:
+                    llm_config["model"] = self.llm_model
             else:
                 llm_config = {
                     "type": "xache-managed",
-                    "model": self.llm_model,
+                    "provider": "anthropic",
                 }
+                if self.llm_model:
+                    llm_config["model"] = self.llm_model
 
             result = await client.extraction.extract(
                 trace=trace,