langchain-xache 0.1.1__tar.gz → 0.2.1__tar.gz

{langchain_xache-0.1.1 → langchain_xache-0.2.1}/PKG-INFO

@@ -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 → langchain_xache-0.2.1}/README.md

@@ -159,9 +159,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 → langchain_xache-0.2.1}/langchain_xache.egg-info/PKG-INFO

@@ -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 → langchain_xache-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "langchain-xache"
-version = "0.1.1"
+version = "0.2.1"
 description = "LangChain integration for Xache Protocol - verifiable AI agent memory"
 readme = "README.md"
 license = {text = "MIT"}

{langchain_xache-0.1.1 → langchain_xache-0.2.1}/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,