neurocli-sdk 0.2.1__tar.gz

neurocli_sdk-0.2.1/PKG-INFO

@@ -0,0 +1,61 @@
+Metadata-Version: 2.4
+Name: neurocli-sdk
+Version: 0.2.1
+Summary: The official Python SDK for the NeuroCLI API
+Home-page: https://www.neurocli.in
+Author: NeuroCLI
+Author-email: NeuroCLI <info@neurocli.in>
+License: MIT
+Project-URL: Homepage, https://www.neurocli.in
+Project-URL: Documentation, https://www.neurocli.in/docs
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Intended Audience :: Developers
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+Requires-Dist: httpx>=0.24.0
+Requires-Dist: tenacity>=8.0.0
+Requires-Dist: pydantic>=2.0.0
+Dynamic: author
+Dynamic: home-page
+Dynamic: requires-python
+
+# NeuroCLI Python SDK
+
+The official Python library for the NeuroCLI API. 
+
+## Installation
+
+You can install this SDK locally:
+```bash
+pip install -e .
+```
+
+## Usage
+
+The SDK mimics the standard OpenAI Python client interface, making it extremely easy to drop into existing AI applications.
+
+### Basic Chat Completion
+
+```python
+import neurocli
+
+# Initialize the client
+client = neurocli.NeuroCLI(
+    api_key="ncli_YOUR_SECRET_KEY",
+    # During local testing, point this to your local server:
+    # base_url="http://127.0.0.1:8000/api/ai" 
+)
+
+# Generate a response
+response = client.chat.completions.create(
+    model="meta/llama-3.1-8b-instruct",
+    messages=[
+        {"role": "user", "content": "Explain quantum computing in simple terms."}
+    ],
+    temperature=0.7
+)
+
+print(response.choices[0].message.content)
+```

neurocli_sdk-0.2.1/README.md

@@ -0,0 +1,38 @@
+# NeuroCLI Python SDK
+
+The official Python library for the NeuroCLI API. 
+
+## Installation
+
+You can install this SDK locally:
+```bash
+pip install -e .
+```
+
+## Usage
+
+The SDK mimics the standard OpenAI Python client interface, making it extremely easy to drop into existing AI applications.
+
+### Basic Chat Completion
+
+```python
+import neurocli
+
+# Initialize the client
+client = neurocli.NeuroCLI(
+    api_key="ncli_YOUR_SECRET_KEY",
+    # During local testing, point this to your local server:
+    # base_url="http://127.0.0.1:8000/api/ai" 
+)
+
+# Generate a response
+response = client.chat.completions.create(
+    model="meta/llama-3.1-8b-instruct",
+    messages=[
+        {"role": "user", "content": "Explain quantum computing in simple terms."}
+    ],
+    temperature=0.7
+)
+
+print(response.choices[0].message.content)
+```

neurocli_sdk-0.2.1/neurocli/__init__.py

@@ -0,0 +1,3 @@
+from .client import NeuroCLI, AsyncNeuroCLI, wrap_function
+
+__all__ = ["NeuroCLI", "AsyncNeuroCLI", "wrap_function"]

neurocli_sdk-0.2.1/neurocli/client.py

@@ -0,0 +1,234 @@
+import json
+import inspect
+from typing import List, Dict, Optional, Any, Callable, Union, Iterator, AsyncIterator
+import httpx
+from pydantic import BaseModel, Field
+from tenacity import retry, stop_after_attempt, wait_exponential, retry_if_exception_type
+
+# --- Types ---
+class Message(BaseModel):
+    role: str
+    content: str
+
+class Choice(BaseModel):
+    message: Message
+    finish_reason: Optional[str] = None
+
+class ChatCompletionResponse(BaseModel):
+    id: Optional[str] = None
+    choices: List[Choice]
+
+# Streaming chunk model
+class ChoiceDelta(BaseModel):
+    content: Optional[str] = None
+
+class ChunkChoice(BaseModel):
+    delta: ChoiceDelta
+    finish_reason: Optional[str] = None
+
+class ChatCompletionChunk(BaseModel):
+    id: Optional[str] = None
+    choices: List[ChunkChoice]
+
+# --- Error Handling ---
+class NeuroCLIError(Exception):
+    pass
+
+class RateLimitError(NeuroCLIError):
+    pass
+
+class APIError(NeuroCLIError):
+    pass
+
+# Retry decorator for rate limits (429) or temporary server errors (502, 503, 504)
+def create_retry_decorator():
+    return retry(
+        stop=stop_after_attempt(3),
+        wait=wait_exponential(multiplier=1, min=2, max=10),
+        retry=retry_if_exception_type((RateLimitError, httpx.ReadTimeout, httpx.ConnectError)),
+        reraise=True
+    )
+
+# --- Function Calling Wrapper ---
+def wrap_function(func: Callable) -> Dict[str, Any]:
+    """
+    Automatically converts a Python function with docstrings and type hints
+    into an OpenAI/Nvidia NIM compatible JSON Schema for function calling.
+    """
+    sig = inspect.signature(func)
+    doc = inspect.getdoc(func) or ""
+    
+    properties = {}
+    required = []
+    
+    for name, param in sig.parameters.items():
+        param_type = "string"
+        if param.annotation == int:
+            param_type = "integer"
+        elif param.annotation == float:
+            param_type = "number"
+        elif param.annotation == bool:
+            param_type = "boolean"
+        
+        properties[name] = {"type": param_type}
+        
+        if param.default == inspect.Parameter.empty:
+            required.append(name)
+            
+    return {
+        "type": "function",
+        "function": {
+            "name": func.__name__,
+            "description": doc,
+            "parameters": {
+                "type": "object",
+                "properties": properties,
+                "required": required
+            }
+        }
+    }
+
+
+# --- Core Logic ---
+
+def _handle_response(response: httpx.Response) -> ChatCompletionResponse:
+    if not response.is_success:
+        if response.status_code == 429:
+            raise RateLimitError("Rate Limit Exceeded. Retrying automatically...")
+        try:
+            error_data = response.json()
+            error_msg = error_data.get('message') or error_data.get('error') or response.text
+        except Exception:
+            error_msg = response.text
+        raise APIError(f"NeuroCLI API Error ({response.status_code}): {error_msg}")
+    
+    data = response.json()
+    return ChatCompletionResponse(**data)
+
+def _stream_iterator(response: httpx.Response) -> Iterator[ChatCompletionChunk]:
+    if not response.is_success:
+        if response.status_code == 429:
+            raise RateLimitError("Rate Limit Exceeded. Retrying automatically...")
+        raise APIError(f"NeuroCLI API Error ({response.status_code})")
+        
+    for line in response.iter_lines():
+        if line.startswith("data: ") and line != "data: [DONE]":
+            data = json.loads(line[6:])
+            yield ChatCompletionChunk(**data)
+
+async def _astream_iterator(response: httpx.Response) -> AsyncIterator[ChatCompletionChunk]:
+    if not response.is_success:
+        if response.status_code == 429:
+            raise RateLimitError("Rate Limit Exceeded. Retrying automatically...")
+        raise APIError(f"NeuroCLI API Error ({response.status_code})")
+        
+    async for line in response.aiter_lines():
+        if line.startswith("data: ") and line != "data: [DONE]":
+            data = json.loads(line[6:])
+            yield ChatCompletionChunk(**data)
+
+
+# --- Sync Client ---
+class ChatCompletions:
+    def __init__(self, client):
+        self._client = client
+
+    @create_retry_decorator()
+    def create(
+        self,
+        model: str,
+        messages: List[Dict[str, str]],
+        temperature: Optional[float] = None,
+        max_tokens: Optional[int] = None,
+        stream: bool = False,
+        tools: Optional[List[Dict[str, Any]]] = None,
+        **kwargs
+    ) -> Union[ChatCompletionResponse, Iterator[ChatCompletionChunk]]:
+        
+        payload = {"model": model, "messages": messages}
+        if temperature is not None: payload["temperature"] = temperature
+        if max_tokens is not None: payload["max_tokens"] = max_tokens
+        if tools is not None: payload["tools"] = tools
+        if stream: payload["stream"] = True
+        payload.update(kwargs)
+
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {self._client.api_key}"
+        }
+
+        if stream:
+            request = self._client._http.build_request(
+                "POST", f"{self._client.base_url}/chat/completions", json=payload, headers=headers
+            )
+            response = self._client._http.send(request, stream=True)
+            return _stream_iterator(response)
+        else:
+            response = self._client._http.post(
+                f"{self._client.base_url}/chat/completions", json=payload, headers=headers
+            )
+            return _handle_response(response)
+
+class Chat:
+    def __init__(self, client):
+        self.completions = ChatCompletions(client)
+
+class NeuroCLI:
+    def __init__(self, api_key: str, base_url: str = "https://www.neurocli.in/v1"):
+        self.api_key = api_key
+        self.base_url = base_url.rstrip("/")
+        self._http = httpx.Client(timeout=60.0)
+        self.chat = Chat(self)
+
+
+# --- Async Client ---
+class AsyncChatCompletions:
+    def __init__(self, client):
+        self._client = client
+
+    @create_retry_decorator()
+    async def create(
+        self,
+        model: str,
+        messages: List[Dict[str, str]],
+        temperature: Optional[float] = None,
+        max_tokens: Optional[int] = None,
+        stream: bool = False,
+        tools: Optional[List[Dict[str, Any]]] = None,
+        **kwargs
+    ) -> Union[ChatCompletionResponse, AsyncIterator[ChatCompletionChunk]]:
+        
+        payload = {"model": model, "messages": messages}
+        if temperature is not None: payload["temperature"] = temperature
+        if max_tokens is not None: payload["max_tokens"] = max_tokens
+        if tools is not None: payload["tools"] = tools
+        if stream: payload["stream"] = True
+        payload.update(kwargs)
+
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {self._client.api_key}"
+        }
+
+        if stream:
+            request = self._client._http.build_request(
+                "POST", f"{self._client.base_url}/chat/completions", json=payload, headers=headers
+            )
+            response = await self._client._http.send(request, stream=True)
+            return _astream_iterator(response)
+        else:
+            response = await self._client._http.post(
+                f"{self._client.base_url}/chat/completions", json=payload, headers=headers
+            )
+            return _handle_response(response)
+
+class AsyncChat:
+    def __init__(self, client):
+        self.completions = AsyncChatCompletions(client)
+
+class AsyncNeuroCLI:
+    def __init__(self, api_key: str, base_url: str = "https://www.neurocli.in/v1"):
+        self.api_key = api_key
+        self.base_url = base_url.rstrip("/")
+        self._http = httpx.AsyncClient(timeout=60.0)
+        self.chat = AsyncChat(self)

neurocli_sdk-0.2.1/neurocli_sdk.egg-info/PKG-INFO

@@ -0,0 +1,61 @@
+Metadata-Version: 2.4
+Name: neurocli-sdk
+Version: 0.2.1
+Summary: The official Python SDK for the NeuroCLI API
+Home-page: https://www.neurocli.in
+Author: NeuroCLI
+Author-email: NeuroCLI <info@neurocli.in>
+License: MIT
+Project-URL: Homepage, https://www.neurocli.in
+Project-URL: Documentation, https://www.neurocli.in/docs
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Intended Audience :: Developers
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+Requires-Dist: httpx>=0.24.0
+Requires-Dist: tenacity>=8.0.0
+Requires-Dist: pydantic>=2.0.0
+Dynamic: author
+Dynamic: home-page
+Dynamic: requires-python
+
+# NeuroCLI Python SDK
+
+The official Python library for the NeuroCLI API. 
+
+## Installation
+
+You can install this SDK locally:
+```bash
+pip install -e .
+```
+
+## Usage
+
+The SDK mimics the standard OpenAI Python client interface, making it extremely easy to drop into existing AI applications.
+
+### Basic Chat Completion
+
+```python
+import neurocli
+
+# Initialize the client
+client = neurocli.NeuroCLI(
+    api_key="ncli_YOUR_SECRET_KEY",
+    # During local testing, point this to your local server:
+    # base_url="http://127.0.0.1:8000/api/ai" 
+)
+
+# Generate a response
+response = client.chat.completions.create(
+    model="meta/llama-3.1-8b-instruct",
+    messages=[
+        {"role": "user", "content": "Explain quantum computing in simple terms."}
+    ],
+    temperature=0.7
+)
+
+print(response.choices[0].message.content)
+```

neurocli_sdk-0.2.1/neurocli_sdk.egg-info/SOURCES.txt

@@ -0,0 +1,10 @@
+README.md
+pyproject.toml
+setup.py
+neurocli/__init__.py
+neurocli/client.py
+neurocli_sdk.egg-info/PKG-INFO
+neurocli_sdk.egg-info/SOURCES.txt
+neurocli_sdk.egg-info/dependency_links.txt
+neurocli_sdk.egg-info/requires.txt
+neurocli_sdk.egg-info/top_level.txt

neurocli_sdk-0.2.1/neurocli_sdk.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

neurocli_sdk-0.2.1/neurocli_sdk.egg-info/requires.txt

@@ -0,0 +1,3 @@
+httpx>=0.24.0
+tenacity>=8.0.0
+pydantic>=2.0.0

neurocli_sdk-0.2.1/neurocli_sdk.egg-info/top_level.txt

@@ -0,0 +1 @@
+neurocli

neurocli_sdk-0.2.1/pyproject.toml

@@ -0,0 +1,29 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "neurocli-sdk"
+version = "0.2.1"
+description = "The official Python SDK for the NeuroCLI API"
+readme = "README.md"
+requires-python = ">=3.7"
+license = { text = "MIT" }
+authors = [
+  { name="NeuroCLI", email="info@neurocli.in" },
+]
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+    "Intended Audience :: Developers",
+]
+dependencies = [
+    "httpx>=0.24.0",
+    "tenacity>=8.0.0",
+    "pydantic>=2.0.0",
+]
+
+[project.urls]
+"Homepage" = "https://www.neurocli.in"
+"Documentation" = "https://www.neurocli.in/docs"

neurocli_sdk-0.2.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

neurocli_sdk-0.2.1/setup.py

@@ -0,0 +1,28 @@
+from setuptools import setup, find_packages
+
+with open("README.md", "r", encoding="utf-8") as fh:
+    long_description = fh.read()
+
+setup(
+    name="neurocli-sdk",
+    version="0.2.1",
+    description="The official Python SDK for the NeuroCLI API.",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    author="NeuroCLI",
+    author_email="info@neurocli.in",
+    url="https://www.neurocli.in",
+    packages=find_packages(),
+    install_requires=[
+        "httpx>=0.24.0",
+        "tenacity>=8.0.0",
+        "pydantic>=2.0.0"
+    ],
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
+        "Intended Audience :: Developers",
+    ],
+    python_requires=">=3.7",
+)