euriai 0.3.3__py3-none-any.whl → 0.3.4__py3-none-any.whl

euriai/__init__.py

@@ -1,4 +1,6 @@
 from .client import EuriaiClient
 from .langchain_llm import EuriaiLangChainLLM
+from .embedding import EuriaiEmbeddingClient
+from .langchain_embed import EuriaiEmbeddings
 
-__all__ = ["EuriaiClient", "EuriaiLangChainLLM"]
+__all__ = ["EuriaiClient", "EuriaiLangChainLLM","EuriaiEmbeddingClient", "EuriaiEmbeddings"]

euriai/embedding.py

@@ -0,0 +1,40 @@
+import requests
+import numpy as np
+
+class EuriaiEmbeddingClient:
+    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 embed(self, text: str) -> np.ndarray:
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {self.api_key}"
+        }
+        payload = {
+            "input": text,
+            "model": self.model
+        }
+
+        response = requests.post(self.url, headers=headers, json=payload)
+        response.raise_for_status()
+        data = response.json()
+
+        return np.array(data["data"][0]["embedding"])
+
+    def embed_batch(self, texts: list[str]) -> list[np.ndarray]:
+        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()
+        data = response.json()
+
+        return [np.array(obj["embedding"]) for obj in data["data"]]

euriai/langchain_embed.py

@@ -0,0 +1,14 @@
+from langchain_core.embeddings import Embeddings
+from typing import List
+from euriai.embedding import EuriaiEmbeddingClient
+
+
+class EuriaiEmbeddings(Embeddings):
+    def __init__(self, api_key: str, model: str = "text-embedding-3-small"):
+        self.client = EuriaiEmbeddingClient(api_key=api_key, model=model)
+
+    def embed_documents(self, texts: List[str]) -> List[List[float]]:
+        return [embedding.tolist() for embedding in self.client.embed_batch(texts)]
+
+    def embed_query(self, text: str) -> List[float]:
+        return self.client.embed(text).tolist()

{euriai-0.3.3.dist-info → euriai-0.3.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: euriai
-Version: 0.3.3
+Version: 0.3.4
 Summary: Python client for EURI LLM API (euron.one) with CLI and interactive wizard
 Author: euron.one
 Author-email: sudhanshu@euron.one
@@ -13,6 +13,15 @@ Requires-Python: >=3.6
 Description-Content-Type: text/markdown
 Requires-Dist: requests
 Requires-Dist: langchain-core
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: license
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 # euriai 🧠
 

euriai-0.3.4.dist-info/RECORD

@@ -0,0 +1,11 @@
+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.4.dist-info/METADATA,sha256=QrZomCuobKBoKfkd1-jvAKcGxzuyf8IKaGQ0va-aW9g,2105
+euriai-0.3.4.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
+euriai-0.3.4.dist-info/entry_points.txt,sha256=9OkET8KIGcsjQn8UlnpPKRT75s2KW34jq1__1SXtpMA,43
+euriai-0.3.4.dist-info/top_level.txt,sha256=TG1htJ8cuD62MXn-NJ7DVF21QHY16w6M_QgfF_Er_EQ,7
+euriai-0.3.4.dist-info/RECORD,,

{euriai-0.3.3.dist-info → euriai-0.3.4.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.1.0)
+Generator: setuptools (80.4.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

euriai-0.3.3.dist-info/RECORD

@@ -1,9 +0,0 @@
-euriai/__init__.py,sha256=MB58A0AUXMvhW9gJWK8NSpnvtGawlejvtKJ1DOhs1SM,133
-euriai/cli.py,sha256=hF1wiiL2QQSfWf8WlLQyNVDBd4YkbiwmMSoPxVbyPTM,3290
-euriai/client.py,sha256=USiqdMULgAiky7nkrJKF3FyKcOS2DtDmUdbeBSnyLYk,4076
-euriai/langchain_llm.py,sha256=D5YvYwV7q9X2_vdoaQiPs7tNiUmjkGz-9Q-7M61hhkg,986
-euriai-0.3.3.dist-info/METADATA,sha256=IHFMzYdAvNb5mROV7ozEpuA4gr8IkStYo11lx0JPEUA,1901
-euriai-0.3.3.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-euriai-0.3.3.dist-info/entry_points.txt,sha256=9OkET8KIGcsjQn8UlnpPKRT75s2KW34jq1__1SXtpMA,43
-euriai-0.3.3.dist-info/top_level.txt,sha256=TG1htJ8cuD62MXn-NJ7DVF21QHY16w6M_QgfF_Er_EQ,7
-euriai-0.3.3.dist-info/RECORD,,