euriai 0.3.0__tar.gz → 0.3.2__tar.gz

{euriai-0.3.0 → euriai-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: euriai
-Version: 0.3.0
+Version: 0.3.2
 Summary: Python client for EURI LLM API (euron.one) with CLI and interactive wizard
 Author: euron.one
 Author-email: sudhanshu@euron.one

euriai-0.3.2/euriai/langchain_llm.py

@@ -0,0 +1,29 @@
+from langchain_core.language_models.llms import LLM
+from typing import Optional, List
+from pydantic import PrivateAttr
+from .client import EuriaiClient
+
+
+class EuriaiLangChainLLM(LLM):
+    model: str = "gpt-4.1-nano"
+    temperature: float = 0.7
+    max_tokens: int = 300
+
+    _client: EuriaiClient = PrivateAttr()
+
+    def __init__(self, api_key: str, model: str = "gpt-4.1-nano", temperature: float = 0.7, max_tokens: int = 300):
+        super().__init__(model=model, temperature=temperature, max_tokens=max_tokens)
+        self._client = EuriaiClient(api_key=api_key, model=model)
+
+    @property
+    def _llm_type(self) -> str:
+        return "euriai"
+
+    def _call(self, prompt: str, stop: Optional[List[str]] = None) -> str:
+        response = self._client.generate_completion(
+            prompt=prompt,
+            temperature=self.temperature,
+            max_tokens=self.max_tokens,
+            stop=stop
+        )
+        return response.get("choices", [{}])[0].get("message", {}).get("content", "")

{euriai-0.3.0 → euriai-0.3.2}/euriai.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: euriai
-Version: 0.3.0
+Version: 0.3.2
 Summary: Python client for EURI LLM API (euron.one) with CLI and interactive wizard
 Author: euron.one
 Author-email: sudhanshu@euron.one

{euriai-0.3.0 → euriai-0.3.2}/euriai.egg-info/SOURCES.txt

@@ -3,6 +3,7 @@ setup.py
 euriai/__init__.py
 euriai/cli.py
 euriai/client.py
+euriai/langchain_llm.py
 euriai.egg-info/PKG-INFO
 euriai.egg-info/SOURCES.txt
 euriai.egg-info/dependency_links.txt

{euriai-0.3.0 → euriai-0.3.2}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="euriai",
-    version="0.3.0",
+    version="0.3.2",
     description="Python client for EURI LLM API (euron.one) with CLI and interactive wizard",
     long_description=open("README.md", encoding="utf-8").read(),
     long_description_content_type="text/markdown",