euriai 0.3.26__tar.gz → 0.3.28__tar.gz

{euriai-0.3.26 → euriai-0.3.28}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: euriai
-Version: 0.3.26
+Version: 0.3.28
 Summary: Python client for EURI LLM API (euron.one) with CLI, LangChain, and LlamaIndex integration
 Author: euron.one
 Author-email: sudhanshu@euron.one

{euriai-0.3.26 → euriai-0.3.28}/euriai/euri_chat.py

@@ -1,10 +1,11 @@
 import requests
-from typing import List, Optional
+from typing import List, Optional, Any, Dict
 from llama_index.core.llms import LLM
 from llama_index.core.base.llms.types import ChatMessage, CompletionResponse, CompletionResponseGen
 
 
 class EuriaiLlamaIndexLLM(LLM):
+    # Define class attributes as expected by Pydantic
     api_key: str
     model: str = "gpt-4.1-nano"
     temperature: float = 0.7
@@ -18,16 +19,28 @@ class EuriaiLlamaIndexLLM(LLM):
         temperature: Optional[float] = None,
         max_tokens: Optional[int] = None,
     ):
-        # ✅ Pydantic-style super init with all fields
-        super().__init__(
-            api_key=api_key,
-            model=model or self.model,
-            temperature=temperature if temperature is not None else self.temperature,
-            max_tokens=max_tokens if max_tokens is not None else self.max_tokens,
-        )
+        """
+        Initializes the EuriaiLlamaIndexLLM.
+
+        Args:
+            api_key (str): Your EURI API key.
+            model (str, optional): Model ID to use. Defaults to "gpt-4.1-nano".
+            temperature (float, optional): Sampling temperature. Defaults to 0.7.
+            max_tokens (int, optional): Maximum number of tokens. Defaults to 1000.
+        """
+        # Create a dictionary of parameters for the parent class
+        model_params = {
+            "api_key": api_key,
+            "model": model if model is not None else self.model,
+            "temperature": temperature if temperature is not None else self.temperature,
+            "max_tokens": max_tokens if max_tokens is not None else self.max_tokens,
+        }
+        
+        # Initialize the parent class with the parameters
+        super().__init__(**model_params)
 
     @property
-    def metadata(self):
+    def metadata(self) -> Dict[str, Any]:
         return {
             "context_window": 8000,
             "num_output": self.max_tokens,

{euriai-0.3.26 → euriai-0.3.28}/euriai/euri_embed.py

@@ -1,15 +1,24 @@
 import requests
 import numpy as np
-from llama_index.core.embeddings import BaseEmbedding  # Correct import path
+from typing import List, Optional
+from llama_index.core.embeddings import BaseEmbedding
 
 class EuriaiLlamaIndexEmbedding(BaseEmbedding):
-    def __init__(self, api_key: str, model: str = "text-embedding-3-small"):
+    # Define class attributes as expected by Pydantic
+    api_key: str
+    model: str = "text-embedding-3-small"
+    url: str = "https://api.euron.one/api/v1/euri/alpha/embeddings"
+
+    def __init__(self, api_key: str, model: Optional[str] = None):
         """Initialize embedding model with API key and model name."""
-        self.api_key = api_key
-        self.model = model
-        self.url = "https://api.euron.one/api/v1/euri/alpha/embeddings"
-        # Call the parent class constructor (important!)
-        super().__init__()
+        # Create parameters for the parent class
+        embed_params = {
+            "api_key": api_key,
+            "model": model if model is not None else self.model,
+        }
+        
+        # Initialize the parent class
+        super().__init__(**embed_params)
 
     def _post_embedding(self, texts):
         """Helper method to post data to API and get embeddings."""
@@ -25,10 +34,10 @@ class EuriaiLlamaIndexEmbedding(BaseEmbedding):
         response.raise_for_status()
         return [np.array(obj["embedding"]).tolist() for obj in response.json()["data"]]
 
-    def get_text_embedding(self, text: str) -> list[float]:
+    def get_text_embedding(self, text: str) -> List[float]:
         """Get embedding for a single text."""
         return self._post_embedding([text])[0]
 
-    def get_text_embeddings(self, texts: list[str]) -> list[list[float]]:
+    def get_text_embeddings(self, texts: List[str]) -> List[List[float]]:
         """Get embeddings for multiple texts."""
         return self._post_embedding(texts)

{euriai-0.3.26 → euriai-0.3.28}/euriai.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: euriai
-Version: 0.3.26
+Version: 0.3.28
 Summary: Python client for EURI LLM API (euron.one) with CLI, LangChain, and LlamaIndex integration
 Author: euron.one
 Author-email: sudhanshu@euron.one

{euriai-0.3.26 → euriai-0.3.28}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="euriai",
-    version="0.3.26",
+    version="0.3.28",
     description="Python client for EURI LLM API (euron.one) with CLI, LangChain, and LlamaIndex integration",
     long_description=open("README.md", encoding="utf-8").read(),
     long_description_content_type="text/markdown",