euriai 0.3.12__tar.gz → 0.3.14__tar.gz

{euriai-0.3.12 → euriai-0.3.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: euriai
-Version: 0.3.12
+Version: 0.3.14
 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.14/euriai/llamaindex/euri_chat.py

@@ -0,0 +1,64 @@
+import requests
+from typing import List
+from dataclasses import dataclass, field
+from llama_index.core.llms import LLM
+from llama_index.core.base.llms.types import ChatMessage, CompletionResponse, CompletionResponseGen
+
+
+@dataclass
+class EuriaiLlamaIndexLLM(LLM):
+    api_key: str
+    model: str = "gpt-4.1-nano"
+    temperature: float = 0.7
+    max_tokens: int = 1000
+    url: str = field(default="https://api.euron.one/api/v1/chat/completions", init=False)
+
+    @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,
+        }
+
+        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 & streaming not implemented
+    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.")
+
+    def stream_complete(self, prompt: str, **kwargs) -> CompletionResponseGen:
+        raise NotImplementedError("Streaming not supported.")
+
+    async def astream_chat(self, messages: List[ChatMessage], **kwargs) -> CompletionResponseGen:
+        raise NotImplementedError("Async streaming not supported.")
+
+    async def astream_complete(self, prompt: str, **kwargs) -> CompletionResponseGen:
+        raise NotImplementedError("Async streaming not supported.")

{euriai-0.3.12 → euriai-0.3.14}/euriai.egg-info/PKG-INFO

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

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

euriai-0.3.12/euriai/llamaindex/euri_chat.py

@@ -1,33 +0,0 @@
-import requests
-from llama_index.core.llms import LLM
-from llama_index.core.base.llms.types import ChatMessage, CompletionResponse
-from typing import List
-
-
-class EuriaiLlamaIndexLLM(LLM):
-    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)])
-
-    def chat(self, messages: List[ChatMessage], **kwargs) -> CompletionResponse:
-        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}"
-        }
-        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)