langchain-zunno 0.1.0__tar.gz → 0.1.1__tar.gz

{langchain_zunno-0.1.0 → langchain_zunno-0.1.1}/PKG-INFO

@@ -1,9 +1,7 @@
 Metadata-Version: 2.4
 Name: langchain-zunno
-Version: 0.1.0
+Version: 0.1.1
 Summary: LangChain integration for Zunno LLM and Embeddings
-Home-page: https://github.com/zunno/langchain-zunno
-Author: Amit Kumar
 Author-email: Amit Kumar <amit@zunno.ai>
 Maintainer-email: Amit Kumar <amit@zunno.ai>
 License: MIT
@@ -38,10 +36,7 @@ Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
 Requires-Dist: black>=22.0.0; extra == "dev"
 Requires-Dist: isort>=5.0.0; extra == "dev"
 Requires-Dist: flake8>=4.0.0; extra == "dev"
-Dynamic: author
-Dynamic: home-page
 Dynamic: license-file
-Dynamic: requires-python
 
 # LangChain Zunno Integration
 

{langchain_zunno-0.1.0 → langchain_zunno-0.1.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "langchain-zunno"
-version = "0.1.0"
+version = "0.1.1"
 description = "LangChain integration for Zunno LLM and Embeddings"
 readme = "README.md"
 license = {text = "MIT"}
@@ -54,8 +54,7 @@ Repository = "https://github.com/zunno/langchain-zunno"
 "Bug Tracker" = "https://github.com/zunno/langchain-zunno/issues"
 
 [tool.setuptools.packages.find]
-where = ["."]
-include = ["langchain_zunno*"]
+where = ["src"]
 
 [tool.black]
 line-length = 88

langchain_zunno-0.1.1/src/langchain_zunno/__init__.py

@@ -0,0 +1,19 @@
+"""
+LangChain Zunno Integration
+
+A LangChain integration for Zunno LLM and Embeddings.
+"""
+
+from .llm import ZunnoLLM, create_zunno_llm
+from .embeddings import ZunnoLLMEmbeddings, create_zunno_embeddings
+
+__version__ = "0.1.0"
+__author__ = "Amit Kumar"
+__email__ = "amit@zunno.ai"
+
+__all__ = [
+    "ZunnoLLM",
+    "ZunnoLLMEmbeddings", 
+    "create_zunno_llm",
+    "create_zunno_embeddings",
+] 

langchain_zunno-0.1.1/src/langchain_zunno/embeddings.py

@@ -0,0 +1,115 @@
+"""
+LangChain Embeddings integration for Zunno.
+"""
+
+from typing import List
+from langchain.embeddings.base import Embeddings
+import requests
+
+
+class ZunnoLLMEmbeddings(Embeddings):
+    """Zunno Embeddings wrapper for LangChain."""
+    
+    model_name: str
+    base_url: str = "http://15.206.124.44/v1/text-embeddings"
+    timeout: int = 300
+    
+    def __init__(self, model_name: str, base_url: str = None, timeout: int = 300):
+        super().__init__()
+        self.model_name = model_name
+        if base_url:
+            self.base_url = base_url
+        self.timeout = timeout
+    
+    def embed_documents(self, texts: List[str]) -> List[List[float]]:
+        """Embed a list of documents."""
+        embeddings = []
+        for text in texts:
+            embedding = self.embed_query(text)
+            embeddings.append(embedding)
+        return embeddings
+    
+    def embed_query(self, text: str) -> List[float]:
+        """Embed a single query."""
+        try:
+            payload = {
+                "model_name": self.model_name,
+                "text": text,
+                "options": {
+                    "normalize": True
+                }
+            }
+            
+            response = requests.post(
+                self.base_url,
+                json=payload,
+                timeout=self.timeout
+            )
+            
+            if response.status_code != 200:
+                raise Exception(f"Zunno embeddings API error: {response.text}")
+            
+            data = response.json()
+            embeddings = data.get("embeddings", [])
+            
+            if not embeddings:
+                raise Exception("No embeddings returned from API")
+                
+            return embeddings
+            
+        except Exception as e:
+            raise Exception(f"Error getting text embeddings: {e}")
+    
+    async def aembed_documents(self, texts: List[str]) -> List[List[float]]:
+        """Async embed a list of documents."""
+        embeddings = []
+        for text in texts:
+            embedding = await self.aembed_query(text)
+            embeddings.append(embedding)
+        return embeddings
+    
+    async def aembed_query(self, text: str) -> List[float]:
+        """Async embed a single query."""
+        try:
+            import httpx
+            
+            async with httpx.AsyncClient(timeout=self.timeout) as client:
+                payload = {
+                    "model_name": self.model_name,
+                    "text": text,
+                    "options": {
+                        "normalize": True
+                    }
+                }
+                
+                response = await client.post(
+                    self.base_url,
+                    json=payload
+                )
+                
+                if response.status_code != 200:
+                    raise Exception(f"Zunno embeddings API error: {response.text}")
+                
+                data = response.json()
+                embeddings = data.get("embeddings", [])
+                
+                if not embeddings:
+                    raise Exception("No embeddings returned from API")
+                    
+                return embeddings
+                
+        except Exception as e:
+            raise Exception(f"Error getting text embeddings: {e}")
+
+
+def create_zunno_embeddings(
+    model_name: str,
+    base_url: str = "http://15.206.124.44/v1/text-embeddings",
+    timeout: int = 300
+) -> ZunnoLLMEmbeddings:
+    """Create a Zunno Embeddings instance."""
+    return ZunnoLLMEmbeddings(
+        model_name=model_name,
+        base_url=base_url,
+        timeout=timeout
+    ) 

langchain_zunno-0.1.1/src/langchain_zunno/llm.py

@@ -0,0 +1,130 @@
+"""
+LangChain LLM integration for Zunno.
+"""
+
+from typing import Any, List, Optional, Dict
+from langchain.llms.base import LLM
+from langchain.callbacks.manager import CallbackManagerForLLMRun
+import requests
+
+
+class ZunnoLLM(LLM):
+    """Zunno LLM wrapper for LangChain."""
+    
+    model_name: str
+    base_url: str = "http://15.206.124.44/v1/prompt-response"
+    temperature: float = 0.7
+    max_tokens: Optional[int] = None
+    timeout: int = 300
+    
+    @property
+    def _llm_type(self) -> str:
+        """Return type of LLM."""
+        return "zunno"
+    
+    @property
+    def _identifying_params(self) -> Dict[str, Any]:
+        """Get the identifying parameters."""
+        return {
+            "model_name": self.model_name,
+            "temperature": self.temperature,
+            "max_tokens": self.max_tokens,
+            "base_url": self.base_url,
+        }
+    
+    def _call(
+        self,
+        prompt: str,
+        stop: Optional[List[str]] = None,
+        run_manager: Optional[CallbackManagerForLLMRun] = None,
+        **kwargs: Any,
+    ) -> str:
+        """Call the Zunno API."""
+        try:
+            payload = {
+                "model_name": self.model_name,
+                "prompt": prompt,
+                "stream": False,
+                "options": {
+                    "temperature": self.temperature,
+                }
+            }
+            
+            if self.max_tokens:
+                payload["options"]["num_predict"] = self.max_tokens
+            
+            if stop:
+                payload["options"]["stop"] = stop
+            
+            response = requests.post(
+                self.base_url,
+                json=payload,
+                timeout=self.timeout
+            )
+            
+            if response.status_code != 200:
+                raise Exception(f"Zunno API error: {response.text}")
+            
+            data = response.json()
+            return data.get("response", "")
+            
+        except Exception as e:
+            raise Exception(f"Error calling Zunno LLM: {e}")
+    
+    async def _acall(
+        self,
+        prompt: str,
+        stop: Optional[List[str]] = None,
+        run_manager: Optional[CallbackManagerForLLMRun] = None,
+        **kwargs: Any,
+    ) -> str:
+        """Async call to the Zunno API."""
+        try:
+            import httpx
+            
+            async with httpx.AsyncClient(timeout=self.timeout) as client:
+                payload = {
+                    "model_name": self.model_name,
+                    "prompt": prompt,
+                    "stream": False,
+                    "options": {
+                        "temperature": self.temperature,
+                    }
+                }
+                
+                if self.max_tokens:
+                    payload["options"]["num_predict"] = self.max_tokens
+                
+                if stop:
+                    payload["options"]["stop"] = stop
+                
+                response = await client.post(
+                    self.base_url,
+                    json=payload
+                )
+
+                if response.status_code != 200:
+                    raise Exception(f"Zunno API error: {response.text}")
+                
+                data = response.json()
+                return data.get("response", "")
+                
+        except Exception as e:
+            raise Exception(f"Error calling Zunno LLM: {e}")
+
+
+def create_zunno_llm(
+    model_name: str,
+    temperature: float = 0.7,
+    max_tokens: Optional[int] = None,
+    base_url: str = "http://15.206.124.44/v1/prompt-response",
+    timeout: int = 300
+) -> ZunnoLLM:
+    """Create a Zunno LLM instance."""
+    return ZunnoLLM(
+        model_name=model_name,
+        temperature=temperature,
+        max_tokens=max_tokens,
+        base_url=base_url,
+        timeout=timeout
+    ) 

{langchain_zunno-0.1.0 → langchain_zunno-0.1.1/src}/langchain_zunno.egg-info/PKG-INFO

@@ -1,9 +1,7 @@
 Metadata-Version: 2.4
 Name: langchain-zunno
-Version: 0.1.0
+Version: 0.1.1
 Summary: LangChain integration for Zunno LLM and Embeddings
-Home-page: https://github.com/zunno/langchain-zunno
-Author: Amit Kumar
 Author-email: Amit Kumar <amit@zunno.ai>
 Maintainer-email: Amit Kumar <amit@zunno.ai>
 License: MIT
@@ -38,10 +36,7 @@ Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
 Requires-Dist: black>=22.0.0; extra == "dev"
 Requires-Dist: isort>=5.0.0; extra == "dev"
 Requires-Dist: flake8>=4.0.0; extra == "dev"
-Dynamic: author
-Dynamic: home-page
 Dynamic: license-file
-Dynamic: requires-python
 
 # LangChain Zunno Integration
 

langchain_zunno-0.1.1/src/langchain_zunno.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+LICENSE
+README.md
+pyproject.toml
+src/langchain_zunno/__init__.py
+src/langchain_zunno/embeddings.py
+src/langchain_zunno/llm.py
+src/langchain_zunno.egg-info/PKG-INFO
+src/langchain_zunno.egg-info/SOURCES.txt
+src/langchain_zunno.egg-info/dependency_links.txt
+src/langchain_zunno.egg-info/requires.txt
+src/langchain_zunno.egg-info/top_level.txt

langchain_zunno-0.1.1/src/langchain_zunno.egg-info/top_level.txt

@@ -0,0 +1 @@
+langchain_zunno

langchain_zunno-0.1.0/langchain_zunno.egg-info/SOURCES.txt

@@ -1,9 +0,0 @@
-LICENSE
-README.md
-pyproject.toml
-setup.py
-langchain_zunno.egg-info/PKG-INFO
-langchain_zunno.egg-info/SOURCES.txt
-langchain_zunno.egg-info/dependency_links.txt
-langchain_zunno.egg-info/requires.txt
-langchain_zunno.egg-info/top_level.txt

langchain_zunno-0.1.0/langchain_zunno.egg-info/top_level.txt

@@ -1 +0,0 @@
-

langchain_zunno-0.1.0/setup.py

@@ -1,53 +0,0 @@
-from setuptools import setup, find_packages
-
-with open("README.md", "r", encoding="utf-8") as fh:
-    long_description = fh.read()
-
-setup(
-    name="langchain-zunno",
-    version="0.1.0",
-    author="Amit Kumar",
-    author_email="amit@zunno.ai",
-    description="LangChain integration for Zunno LLM and Embeddings",
-    long_description=long_description,
-    long_description_content_type="text/markdown",
-    url="https://github.com/zunno/langchain-zunno",
-    packages=find_packages(),
-    classifiers=[
-        "Development Status :: 3 - Alpha",
-        "Intended Audience :: Developers",
-        "License :: OSI Approved :: MIT License",
-        "Operating System :: OS Independent",
-        "Programming Language :: Python :: 3",
-        "Programming Language :: Python :: 3.8",
-        "Programming Language :: Python :: 3.9",
-        "Programming Language :: Python :: 3.10",
-        "Programming Language :: Python :: 3.11",
-        "Programming Language :: Python :: 3.12",
-        "Topic :: Scientific/Engineering :: Artificial Intelligence",
-        "Topic :: Software Development :: Libraries :: Python Modules",
-    ],
-    python_requires=">=3.8",
-    install_requires=[
-        "langchain>=0.1.0",
-        "langchain-core>=0.1.0",
-        "requests>=2.25.0",
-        "httpx>=0.24.0",
-        "pydantic>=2.0.0",
-    ],
-    extras_require={
-        "dev": [
-            "pytest>=7.0.0",
-            "pytest-asyncio>=0.21.0",
-            "black>=22.0.0",
-            "isort>=5.0.0",
-            "flake8>=4.0.0",
-        ],
-    },
-    keywords="langchain, llm, embeddings, zunno, ai, machine-learning",
-    project_urls={
-        "Bug Reports": "https://github.com/zunno/langchain-zunno/issues",
-        "Source": "https://github.com/zunno/langchain-zunno",
-        "Documentation": "https://github.com/zunno/langchain-zunno#readme",
-    },
-)