euriai 0.3.27__py3-none-any.whl → 0.3.28__py3-none-any.whl

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/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.dist-info → euriai-0.3.28.dist-info}/METADATA

@@ -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.dist-info → euriai-0.3.28.dist-info}/RECORD

@@ -2,12 +2,12 @@ euriai/__init__.py,sha256=8YhbRV8s4wmNxDP9KmjOxQYLgKUCke450vnp-8-kPKs,449
 euriai/cli.py,sha256=hF1wiiL2QQSfWf8WlLQyNVDBd4YkbiwmMSoPxVbyPTM,3290
 euriai/client.py,sha256=USiqdMULgAiky7nkrJKF3FyKcOS2DtDmUdbeBSnyLYk,4076
 euriai/embedding.py,sha256=z-LLKU68tCrPi9QMs1tlKwyr7WJcjceCTkNQIFMG6vA,1276
-euriai/euri_chat.py,sha256=LdUxoXVwf5__Lw0GEK4vDph9mGKBYP5749L8tjT8HX4,3273
-euriai/euri_embed.py,sha256=EMyamZ8DtZzxoJsaRPXyOrZWuhP7GgIpKOoMXJeXHUc,1423
+euriai/euri_chat.py,sha256=I2n0ubzGW7-SLp2Kg_fRJ9QLZAP-fY5rSe6azn5QF6c,3585
+euriai/euri_embed.py,sha256=nwqFQ8ozMSP0xiTIf6W1QNur8SzV2OPBJRKNw7J7ouw,1650
 euriai/langchain_embed.py,sha256=OXWWxiKJ4g24TFgnWPOCZvhK7G8xtSf0ppQ2zwHkIPM,584
 euriai/langchain_llm.py,sha256=D5YvYwV7q9X2_vdoaQiPs7tNiUmjkGz-9Q-7M61hhkg,986
-euriai-0.3.27.dist-info/METADATA,sha256=vvXZTLD6a0-fENO3UCCM2c4avddFgdVzWHw9NlH3Hv8,3249
-euriai-0.3.27.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
-euriai-0.3.27.dist-info/entry_points.txt,sha256=9OkET8KIGcsjQn8UlnpPKRT75s2KW34jq1__1SXtpMA,43
-euriai-0.3.27.dist-info/top_level.txt,sha256=TG1htJ8cuD62MXn-NJ7DVF21QHY16w6M_QgfF_Er_EQ,7
-euriai-0.3.27.dist-info/RECORD,,
+euriai-0.3.28.dist-info/METADATA,sha256=30SD7cLiZoHRa7eu-2tOjdIImn3BPEYKLdu-9_gAGZg,3249
+euriai-0.3.28.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
+euriai-0.3.28.dist-info/entry_points.txt,sha256=9OkET8KIGcsjQn8UlnpPKRT75s2KW34jq1__1SXtpMA,43
+euriai-0.3.28.dist-info/top_level.txt,sha256=TG1htJ8cuD62MXn-NJ7DVF21QHY16w6M_QgfF_Er_EQ,7
+euriai-0.3.28.dist-info/RECORD,,