euriai 0.3.11__py3-none-any.whl → 0.3.13__py3-none-any.whl

euriai/llamaindex/euri_chat.py

@@ -1,32 +1,69 @@
 import requests
-from llama_index.core.llms import LLM
-from llama_index.core.llms.types import ChatMessage, CompletionResponse
 from typing import List
+from llama_index.core.llms import LLM
+from llama_index.core.base.llms.types import ChatMessage, CompletionResponse, CompletionResponseGen
+
 
 class EuriaiLlamaIndexLLM(LLM):
-    def __init__(self, api_key: str, model: str = "gpt-4.1-nano", temperature: float = 0.7, max_tokens: int = 1000):
+    def __init__(
+        self,
+        api_key: str,
+        model: str = "gpt-4.1-nano",
+        temperature: float = 0.7,
+        max_tokens: int = 1000,
+    ):
         self.api_key = api_key
         self.model = model
         self.temperature = temperature
         self.max_tokens = max_tokens
         self.url = "https://api.euron.one/api/v1/chat/completions"
 
-    def complete(self, prompt: str, **kwargs) -> CompletionResponse:
-        return self.chat([ChatMessage(role="user", content=prompt)])
+    @property
+    def metadata(self):
+        return {
+            "context_window": 8000,
+            "num_output": self.max_tokens,
+            "is_chat_model": True,
+            "model_name": self.model,
+        }
 
     def chat(self, messages: List[ChatMessage], **kwargs) -> CompletionResponse:
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {self.api_key}",
+        }
         payload = {
             "messages": [{"role": m.role, "content": m.content} for m in messages],
             "model": self.model,
             "temperature": self.temperature,
-            "max_tokens": self.max_tokens
-        }
-        headers = {
-            "Content-Type": "application/json",
-            "Authorization": f"Bearer {self.api_key}"
+            "max_tokens": self.max_tokens,
         }
+
         response = requests.post(self.url, headers=headers, json=payload)
         response.raise_for_status()
         result = response.json()
         content = result["choices"][0]["message"]["content"]
+
         return CompletionResponse(text=content)
+
+    def complete(self, prompt: str, **kwargs) -> CompletionResponse:
+        return self.chat([ChatMessage(role="user", content=prompt)])
+
+    # Async versions (not supported, so raise or return NotImplementedError)
+    async def achat(self, messages: List[ChatMessage], **kwargs) -> CompletionResponse:
+        raise NotImplementedError("Async chat not implemented for Euriai.")
+
+    async def acomplete(self, prompt: str, **kwargs) -> CompletionResponse:
+        raise NotImplementedError("Async complete not implemented for Euriai.")
+
+    def stream_chat(self, messages: List[ChatMessage], **kwargs) -> CompletionResponseGen:
+        raise NotImplementedError("Streaming not supported in EuriaiLlamaIndexLLM.")
+
+    def stream_complete(self, prompt: str, **kwargs) -> CompletionResponseGen:
+        raise NotImplementedError("Streaming not supported in EuriaiLlamaIndexLLM.")
+
+    async def astream_chat(self, messages: List[ChatMessage], **kwargs) -> CompletionResponseGen:
+        raise NotImplementedError("Async streaming not supported in EuriaiLlamaIndexLLM.")
+
+    async def astream_complete(self, prompt: str, **kwargs) -> CompletionResponseGen:
+        raise NotImplementedError("Async streaming not supported in EuriaiLlamaIndexLLM.")

{euriai-0.3.11.dist-info → euriai-0.3.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: euriai
-Version: 0.3.11
+Version: 0.3.13
 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.11.dist-info → euriai-0.3.13.dist-info}/RECORD

@@ -5,10 +5,10 @@ euriai/embedding.py,sha256=z-LLKU68tCrPi9QMs1tlKwyr7WJcjceCTkNQIFMG6vA,1276
 euriai/langchain_embed.py,sha256=OXWWxiKJ4g24TFgnWPOCZvhK7G8xtSf0ppQ2zwHkIPM,584
 euriai/langchain_llm.py,sha256=D5YvYwV7q9X2_vdoaQiPs7tNiUmjkGz-9Q-7M61hhkg,986
 euriai/llamaindex/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-euriai/llamaindex/euri_chat.py,sha256=i9ZxC6J8kNQxY3uJLWxT7OxWQn-c5T3Zg7a2loMrz3w,1372
+euriai/llamaindex/euri_chat.py,sha256=_n3LpslD-QsDKtB3oCMjatgwqmTv7TtKJSvGzCsS0LM,2816
 euriai/llamaindex/euri_embed.py,sha256=RO62uzT9rStp9_Ow2cYa_ZrGabkPFX89ZL3Mgb13X08,1063
-euriai-0.3.11.dist-info/METADATA,sha256=fNdXZnC_k7u3Dm_oftld6KHk03d6DuozNa689kQw-eA,3257
-euriai-0.3.11.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
-euriai-0.3.11.dist-info/entry_points.txt,sha256=9OkET8KIGcsjQn8UlnpPKRT75s2KW34jq1__1SXtpMA,43
-euriai-0.3.11.dist-info/top_level.txt,sha256=TG1htJ8cuD62MXn-NJ7DVF21QHY16w6M_QgfF_Er_EQ,7
-euriai-0.3.11.dist-info/RECORD,,
+euriai-0.3.13.dist-info/METADATA,sha256=u3jmtY4iHdw1P9BVJzxsV-y6OU0EFqgbcJiIkBQbGeU,3257
+euriai-0.3.13.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
+euriai-0.3.13.dist-info/entry_points.txt,sha256=9OkET8KIGcsjQn8UlnpPKRT75s2KW34jq1__1SXtpMA,43
+euriai-0.3.13.dist-info/top_level.txt,sha256=TG1htJ8cuD62MXn-NJ7DVF21QHY16w6M_QgfF_Er_EQ,7
+euriai-0.3.13.dist-info/RECORD,,