euriai 0.3.7__py3-none-any.whl → 0.3.8__py3-none-any.whl

euriai/llamaindex/euri_chat.py

@@ -0,0 +1,31 @@
+import requests
+from llama_index.llms.base import LLM, 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)

euriai/llamaindex/euri_embed.py

@@ -0,0 +1,28 @@
+import requests
+import numpy as np
+from llama_index.embeddings.base import BaseEmbedding
+
+class EuriaiLlamaIndexEmbedding(BaseEmbedding):
+    def __init__(self, api_key: str, model: str = "text-embedding-3-small"):
+        self.api_key = api_key
+        self.model = model
+        self.url = "https://api.euron.one/api/v1/euri/alpha/embeddings"
+
+    def _post_embedding(self, texts):
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {self.api_key}"
+        }
+        payload = {
+            "input": texts,
+            "model": self.model
+        }
+        response = requests.post(self.url, headers=headers, json=payload)
+        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]:
+        return self._post_embedding([text])[0]
+
+    def get_text_embeddings(self, texts: list[str]) -> list[list[float]]:
+        return self._post_embedding(texts)

{euriai-0.3.7.dist-info → euriai-0.3.8.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: euriai
-Version: 0.3.7
-Summary: Python client for EURI LLM API (euron.one) with CLI and interactive wizard
+Version: 0.3.8
+Summary: Python client for EURI LLM API (euron.one) with CLI, LangChain, and LlamaIndex integration
 Author: euron.one
 Author-email: sudhanshu@euron.one
 License: MIT
@@ -13,6 +13,8 @@ Requires-Python: >=3.6
 Description-Content-Type: text/markdown
 Requires-Dist: requests
 Requires-Dist: langchain-core
+Requires-Dist: llama-index
+Requires-Dist: numpy
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

euriai-0.3.8.dist-info/RECORD

@@ -0,0 +1,14 @@
+euriai/__init__.py,sha256=iCrFMfjSnipe5BmOdLdkCCk0dO_BZldCWD79mY67KpE,270
+euriai/cli.py,sha256=hF1wiiL2QQSfWf8WlLQyNVDBd4YkbiwmMSoPxVbyPTM,3290
+euriai/client.py,sha256=USiqdMULgAiky7nkrJKF3FyKcOS2DtDmUdbeBSnyLYk,4076
+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=wvSiNLzvq0Jp8jYif3QDHCfjbodDmQbiuIOIrXEu_Mc,1332
+euriai/llamaindex/euri_embed.py,sha256=HtbjbCBvKka77hpHPlTI1pX28uUL9L3eKbT69rkt4i8,1058
+euriai-0.3.8.dist-info/METADATA,sha256=UNvCyc0K4_ZjH_rPEpOO5ua-srCxb2Z3WEvMu51Jg0c,3240
+euriai-0.3.8.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
+euriai-0.3.8.dist-info/entry_points.txt,sha256=9OkET8KIGcsjQn8UlnpPKRT75s2KW34jq1__1SXtpMA,43
+euriai-0.3.8.dist-info/top_level.txt,sha256=TG1htJ8cuD62MXn-NJ7DVF21QHY16w6M_QgfF_Er_EQ,7
+euriai-0.3.8.dist-info/RECORD,,

euriai-0.3.7.dist-info/RECORD

@@ -1,11 +0,0 @@
-euriai/__init__.py,sha256=iCrFMfjSnipe5BmOdLdkCCk0dO_BZldCWD79mY67KpE,270
-euriai/cli.py,sha256=hF1wiiL2QQSfWf8WlLQyNVDBd4YkbiwmMSoPxVbyPTM,3290
-euriai/client.py,sha256=USiqdMULgAiky7nkrJKF3FyKcOS2DtDmUdbeBSnyLYk,4076
-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-0.3.7.dist-info/METADATA,sha256=7tXdBoErthPhwGELFJ_xfjq_JRNic8HtYZlcyKM58xE,3174
-euriai-0.3.7.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
-euriai-0.3.7.dist-info/entry_points.txt,sha256=9OkET8KIGcsjQn8UlnpPKRT75s2KW34jq1__1SXtpMA,43
-euriai-0.3.7.dist-info/top_level.txt,sha256=TG1htJ8cuD62MXn-NJ7DVF21QHY16w6M_QgfF_Er_EQ,7
-euriai-0.3.7.dist-info/RECORD,,