euriai 0.3.27__tar.gz → 0.3.28__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: euriai
-Version: 0.3.27
+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.27 → euriai-0.3.28}/euriai/euri_chat.py

@@ -1,10 +1,17 @@
 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
+    max_tokens: int = 1000
+    url: str = "https://api.euron.one/api/v1/euri/alpha/chat/completions"
+
     def __init__(
         self,
         api_key: str,
@@ -21,18 +28,19 @@ class EuriaiLlamaIndexLLM(LLM):
             temperature (float, optional): Sampling temperature. Defaults to 0.7.
             max_tokens (int, optional): Maximum number of tokens. Defaults to 1000.
         """
-        # Set default values first
-        self.api_key = api_key
-        self.model = model if model is not None else "gpt-4.1-nano"
-        self.temperature = temperature if temperature is not None else 0.7
-        self.max_tokens = max_tokens if max_tokens is not None else 1000
-        self.url = "https://api.euron.one/api/v1/euri/alpha/chat/completions"
+        # 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,
+        }
         
-        # Now initialize the parent class
-        super().__init__()
+        # 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.27 → euriai-0.3.28}/euriai/euri_embed.py

@@ -1,17 +1,24 @@
 import requests
 import numpy as np
+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."""
-        # Set instance attributes first
-        self.api_key = api_key
-        self.model = model
-        self.url = "https://api.euron.one/api/v1/euri/alpha/embeddings"
+        # Create parameters for the parent class
+        embed_params = {
+            "api_key": api_key,
+            "model": model if model is not None else self.model,
+        }
         
-        # Then call the parent class constructor
-        super().__init__()
+        # Initialize the parent class
+        super().__init__(**embed_params)
 
     def _post_embedding(self, texts):
         """Helper method to post data to API and get embeddings."""
@@ -27,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.27 → euriai-0.3.28}/euriai.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: euriai
-Version: 0.3.27
+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.27 → euriai-0.3.28}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="euriai",
-    version="0.3.27",
+    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",