confamnode 0.1.0__tar.gz → 0.1.1__tar.gz

{confamnode-0.1.0 → confamnode-0.1.1}/.gitignore

@@ -18,3 +18,11 @@ __pycache__/
 dist/
 *.egg-info/
 .pytest_cache/
+
+# Test files
+test_chat.py
+test_stream.py
+exchange_rate_calc.py
+
+# Other files
+main.py

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

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: confamnode
-Version: 0.1.0
-Summary: The Nigerian AI inference SDK — access powerful AI models and pay in Naira
+Version: 0.1.1
+Summary: The Nigerian AI inference gateway — access powerful AI models
 Project-URL: Repository, https://github.com/joteqthefirst/confamnode-sdk
 Project-URL: Bug Tracker, https://github.com/joteqthefirst/confamnode-sdk/issues
 Author-email: JoTeq the First <joteqthefirst@gmail.com>
@@ -24,7 +24,7 @@ Description-Content-Type: text/markdown
 
 # ConfamNode
 
-The Nigerian AI inference SDK — access powerful AI models.
+The Nigerian AI inference gateway — access powerful AI models.
 
 Built by **JoTeq the First**
 
@@ -32,10 +32,44 @@ Built by **JoTeq the First**
 
 ## Installation
 
+### Using pip
 ```bash
 pip install confamnode
 ```
 
+### Using uv
+```bash
+uv add confamnode
+```
+
+### Using virtualenv
+```bash
+# Create virtual environment
+python -m venv .venv
+
+# Activate — Linux/Mac
+source .venv/bin/activate
+
+# Activate — Windows
+.venv\Scripts\activate
+
+# Install
+pip install confamnode
+```
+
+### Using uv with virtual environment
+```bash
+# Create project with uv
+uv init my-project
+cd my-project
+
+# Add confamnode
+uv add confamnode
+
+# Run your script
+uv run python main.py
+```
+
 ---
 
 ## Quick Start
@@ -45,12 +79,12 @@ from confamnode import ConfamNode
 
 client = ConfamNode(api_key="confam-sk-xxx")
 
-response = client.gist(
+ansa = client.gist(
     model="confam-speed",
     messages="How far?"
 )
 
-print(response.choices[0].message.content)
+print(ansa.choices[0].message.content)
 ```
 
 ---
@@ -58,12 +92,12 @@ print(response.choices[0].message.content)
 ## Streaming
 
 ```python
-for chunk in client.gist(
+for yarn in client.gist(
     model="confam-speed",
     messages="How far?",
     stream=True
 ):
-    print(chunk.choices[0].delta.content, end="")
+    print(yarn.choices[0].delta.content, end="")
 ```
 
 ---
@@ -98,23 +132,23 @@ client = ConfamNode()
 from confamnode import ConfamAuthError, ConfamRateLimitError, ConfamModelError
 
 try:
-    response = client.gist(
+    ansa = client.gist(
         model="confam-speed",
         messages="How far?"
     )
 except ConfamAuthError:
-    print("Check your API key")
+    print("You sure say na the correct API Key be that")
 except ConfamRateLimitError:
-    print("You don reach your limit. To upgrade your plan, contact us at confamnode@gmail.com")
+    print("You don reach your limit. Contact us at confamnode@gmail.com make we upgrade your plan")
 except ConfamModelError:
-    print("Invalid model name")
+    print("This model name no dey valid")
 ```
 
 ---
 
 ## Private Deployment
 
-Need NDPA-compliant private AI on your own infrastructure? JoTeq the First offers on-premise deployment on Jetson devices and GPUs, RAG pipelines, fine-tuning, and dedicated hosted models.
+Need NDPA-compliant private AI on your own infrastructure? JoTeq the First offers on-premise deployment on Jetson devices and GPUs, RAG pipelines, fine-tuning, eevaluation, monitoring, and dedicated hosted models.
 
 Contact us at **joteqthefirst@gmail.com**
 

{confamnode-0.1.0 → confamnode-0.1.1}/README.md

@@ -1,6 +1,6 @@
 # ConfamNode
 
-The Nigerian AI inference SDK — access powerful AI models.
+The Nigerian AI inference gateway — access powerful AI models.
 
 Built by **JoTeq the First**
 
@@ -8,10 +8,44 @@ Built by **JoTeq the First**
 
 ## Installation
 
+### Using pip
 ```bash
 pip install confamnode
 ```
 
+### Using uv
+```bash
+uv add confamnode
+```
+
+### Using virtualenv
+```bash
+# Create virtual environment
+python -m venv .venv
+
+# Activate — Linux/Mac
+source .venv/bin/activate
+
+# Activate — Windows
+.venv\Scripts\activate
+
+# Install
+pip install confamnode
+```
+
+### Using uv with virtual environment
+```bash
+# Create project with uv
+uv init my-project
+cd my-project
+
+# Add confamnode
+uv add confamnode
+
+# Run your script
+uv run python main.py
+```
+
 ---
 
 ## Quick Start
@@ -21,12 +55,12 @@ from confamnode import ConfamNode
 
 client = ConfamNode(api_key="confam-sk-xxx")
 
-response = client.gist(
+ansa = client.gist(
     model="confam-speed",
     messages="How far?"
 )
 
-print(response.choices[0].message.content)
+print(ansa.choices[0].message.content)
 ```
 
 ---
@@ -34,12 +68,12 @@ print(response.choices[0].message.content)
 ## Streaming
 
 ```python
-for chunk in client.gist(
+for yarn in client.gist(
     model="confam-speed",
     messages="How far?",
     stream=True
 ):
-    print(chunk.choices[0].delta.content, end="")
+    print(yarn.choices[0].delta.content, end="")
 ```
 
 ---
@@ -74,23 +108,23 @@ client = ConfamNode()
 from confamnode import ConfamAuthError, ConfamRateLimitError, ConfamModelError
 
 try:
-    response = client.gist(
+    ansa = client.gist(
         model="confam-speed",
         messages="How far?"
     )
 except ConfamAuthError:
-    print("Check your API key")
+    print("You sure say na the correct API Key be that")
 except ConfamRateLimitError:
-    print("You don reach your limit. To upgrade your plan, contact us at confamnode@gmail.com")
+    print("You don reach your limit. Contact us at confamnode@gmail.com make we upgrade your plan")
 except ConfamModelError:
-    print("Invalid model name")
+    print("This model name no dey valid")
 ```
 
 ---
 
 ## Private Deployment
 
-Need NDPA-compliant private AI on your own infrastructure? JoTeq the First offers on-premise deployment on Jetson devices and GPUs, RAG pipelines, fine-tuning, and dedicated hosted models.
+Need NDPA-compliant private AI on your own infrastructure? JoTeq the First offers on-premise deployment on Jetson devices and GPUs, RAG pipelines, fine-tuning, eevaluation, monitoring, and dedicated hosted models.
 
 Contact us at **joteqthefirst@gmail.com**
 

{confamnode-0.1.0 → confamnode-0.1.1}/confamnode/__init__.py

@@ -1,20 +1,25 @@
-from confamnode.client import ConfamNode
+from confamnode.client import ConfamNode, ConfamStream
 from confamnode.exceptions import (
     ConfamNodeError,
     ConfamAuthError,
     ConfamRateLimitError,
     ConfamModelError,
 )
+from confamnode.ansa import Ansa, Usage, Cost
 from confamnode import models
 
-__version__ = "0.1.0"
+__version__ = "0.1.1"
 
 __all__ = [
     "ConfamNode",
+    "ConfamStream",
     "ConfamNodeError",
     "ConfamAuthError",
     "ConfamRateLimitError",
     "ConfamModelError",
+    "Ansa",
+    "Usage",
+    "Cost",
     "models",
     "__version__"
 ]

confamnode-0.1.1/confamnode/ansa.py

@@ -0,0 +1,31 @@
+import uuid
+from dataclasses import dataclass, field
+
+
+@dataclass
+class Usage:
+    prompt_tokens: int
+    completion_tokens: int
+    total_tokens: int
+
+
+@dataclass
+class Cost:
+    naira: float
+    naira_input: float = 0.0
+    naira_output: float = 0.0
+    dollars: float | None = None
+
+
+@dataclass
+class Ansa:
+    text: str
+    model: str
+    usage: Usage
+    cost: Cost
+    finish_reason: str
+    raw: object
+    reasoning: str | None = None
+    tools: list = field(default_factory=list)
+    citations: list = field(default_factory=list)
+    id: str = field(default_factory=lambda: f"confam-{uuid.uuid4()}")

confamnode-0.1.1/confamnode/client.py

@@ -0,0 +1,187 @@
+import os
+import litellm
+
+from typing import Union, List, Dict
+
+from confamnode import models
+from confamnode.ansa import Ansa, Usage, Cost
+from confamnode.exceptions import ConfamAuthError, ConfamModelError
+
+VALID_MODELS = [
+    models.LITE,
+    models.SPEED,
+    models.REASONING,
+
+    models.INTELLIGENCE,
+    models.DEEP_REASONING,
+    models.CODE,
+    models.CODE_PRO,
+    models.VISION,
+    models.AUDIO,
+    models.TTS,
+    
+    models.NANO,
+
+    models.EMBED_TEXT,
+    models.EMBED_MULTIMODAL,
+    models.EMBED_MULTIMODAL_2,
+
+    # Paid embeddings
+    models.EMBED_TEXT_PRO,
+    models.EMBED_MULTIMODAL_PRO,
+    models.EMBED_MULTILINGUAL,
+    models.EMBED_SMALL,
+    models.EMBED_0_6B,
+    models.EMBED_TEXT_LOCAL,
+
+    # Rerank
+    models.RERANK,
+    models.RERANK_FAST,
+]
+
+DEFAULT_BASE_URL = "https://api.confamnode.com/v1"
+DEFAULT_USD_TO_NAIRA = 1400.0
+
+class ConfamNode:
+    def __init__(
+        self,
+        api_key: str = None,
+        base_url: str = None
+    ):
+        # Pick up from environment if not provided
+        api_key = api_key or os.environ.get("CONFAMNODE_API_KEY")
+
+        if not api_key:
+            raise ValueError("api_key is required")
+        
+        if not api_key.startswith("confam-sk-"):
+            raise ConfamAuthError()
+        
+        self.api_key = api_key
+        self.litellm_key = api_key.removeprefix("confam-")
+        self.base_url = base_url or DEFAULT_BASE_URL
+
+    def gist(
+        self,
+        model: str,
+        messages: Union[str, List[Dict[str, str]]],
+        **kwargs
+    ) -> object:
+        if model not in VALID_MODELS:
+            raise ConfamModelError(model)
+        
+        if isinstance(messages, str):
+            messages = [{"role": "user", "content": messages}]
+        elif not isinstance(messages, list):
+            raise ValueError("messages must be a string or list")
+        
+        raw = litellm.completion(
+            model=f"openai/{model}",
+            messages=messages,
+            api_key=self.litellm_key,
+            base_url=self.base_url,
+            **kwargs
+        )
+
+        if kwargs.get("stream", False):
+            return ConfamStream(raw, model)
+
+        hidden = getattr(raw, "_hidden_params", {})
+        naira_cost = float(hidden.get("x_confam_naira_cost", 0.0))
+        naira_input = float(hidden.get("x_confam_naira_input", 0.0))
+        naira_output = float(hidden.get("x_confam_naira_output", 0.0))
+        usd_cost = hidden.get("x_confam_usd_cost", None)
+        if usd_cost is not None:
+            usd_cost = float(usd_cost)
+
+        # Extract text
+        text = raw.choices[0].message.content or ""
+
+        # Extract reasoning if present
+        reasoning = getattr(raw.choices[0].message, "reasoning_content", None)
+
+        # Extract tool calls if present
+        tools = []
+        if hasattr(raw.choices[0].message, "tool_calls") and raw.choices[0].message.tool_calls:
+            tools = raw.choices[0].message.tool_calls
+
+        # Extract citations
+        citations = []
+        if hasattr(raw.choices[0].message, "citations") and raw.choices[0].message.citations:
+            citations = raw.choices[0].message.citations
+
+        # Extract usage
+        usage = Usage(
+            prompt_tokens=raw.usage.prompt_tokens,
+            completion_tokens=raw.usage.completion_tokens,
+            total_tokens=raw.usage.total_tokens,
+        )
+
+        cost = Cost(
+            naira=naira_cost,
+            naira_input=naira_input,
+            naira_output=naira_output,
+            dollars=usd_cost
+        )
+
+        return Ansa(
+            text=text,
+            model=model,
+            reasoning=reasoning,
+            tools=tools,
+            citations=citations,
+            usage=usage,
+            cost=cost,
+            finish_reason=raw.choices[0].finish_reason,
+            raw=raw,
+        )
+    
+
+class ConfamStream:
+    def __init__(self, raw_stream, model: str):
+        self._raw_stream = raw_stream
+        self._model = model
+        self._chunks = []
+        self._ansa = None
+
+    def __iter__(self):
+        for yarn in self._raw_stream:
+            self._chunks.append(yarn)
+            yield yarn
+        # Build Ansa after stream completes
+        self._ansa = self._build_ansa()
+
+    def get_ansa(self) -> Ansa:
+        if self._ansa is None:
+            raise RuntimeError("Stream not complete yet. Iterate through all chunks first.")
+        return self._ansa
+    
+    def _build_ansa(self) -> Ansa:
+        # Collect text from all chunks
+        text = "".join([
+            c.choices[0].delta.content or ""
+            for c in self._chunks
+            if c.choices[0].delta.content
+        ])
+
+        # Get finish reason from last chunk
+        finish_reason = self._chunks[-1].choices[0].finish_reason if self._chunks else "stop"
+
+        # Usage — only available in last chunk for some providers
+        last_chunk = self._chunks[-1] if self._chunks else None
+        usage = Usage(
+            prompt_tokens=getattr(getattr(last_chunk, "usage", None), "prompt_tokens", 0),
+            completion_tokens=getattr(getattr(last_chunk, "usage", None), "completion_tokens", 0),
+            total_tokens=getattr(getattr(last_chunk, "usage", None), "total_tokens", 0),
+        )
+
+        cost = Cost(naira=0.0) 
+
+        return Ansa(
+            text=text,
+            model=self._model,
+            usage=usage,
+            cost=cost,
+            finish_reason=finish_reason,
+            raw=self._chunks,
+        )

confamnode-0.1.1/confamnode/models.py

@@ -0,0 +1,45 @@
+# =========================================================================
+# 🆓 FREE TIER — CHAT
+# =========================================================================
+LITE = "confam-lite"
+SPEED = "confam-speed"
+REASONING = "confam-reasoning"
+
+# =========================================================================
+# 💎 PAID TIER — CHAT
+# =========================================================================
+INTELLIGENCE = "confam-intelligence"
+DEEP_REASONING = "confam-deep-reasoning"
+CODE = "confam-code"
+CODE_PRO = "confam-code-pro"
+VISION = "confam-vision"
+AUDIO = "confam-audio"
+TTS = "confam-tts"
+
+# =========================================================================
+# 🛠️ LOCAL PAID (NDPA Compliant)
+# =========================================================================
+NANO = "confam-nano"
+
+# =========================================================================
+# 📦 FREE EMBEDDINGS
+# =========================================================================
+EMBED_TEXT = "confam-embed-text-1"
+EMBED_MULTIMODAL = "confam-embed-multimodal-1"
+EMBED_MULTIMODAL_2 = "confam-embed-multimodal-2"
+
+# =========================================================================
+# 💎 PAID EMBEDDINGS
+# =========================================================================
+EMBED_TEXT_PRO = "confam-embed-text-pro"
+EMBED_MULTIMODAL_PRO = "confam-embed-multimodal-pro"
+EMBED_MULTILINGUAL = "confam-embed-multilingual"
+EMBED_SMALL = "confam-embed-small"
+EMBED_0_6B = "confam-embed-0.6b"
+EMBED_TEXT_LOCAL = "confam-embed-text-local"
+
+# =========================================================================
+# 🔍 RERANK
+# =========================================================================
+RERANK = "confam-rerank"
+RERANK_FAST = "confam-rerank-fast"

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

@@ -1,7 +1,7 @@
 [project]
 name = "confamnode"
-version = "0.1.0"
-description = "The Nigerian AI inference SDK — access powerful AI models and pay in Naira"
+version = "0.1.1"
+description = "The Nigerian AI inference gateway — access powerful AI models"
 readme = "README.md"
 requires-python = ">=3.10"
 license = { text = "MIT" }
@@ -34,6 +34,7 @@ dev = [
     "build>=1.5.0",
     "pytest>=9.0.3",
     "pytest-dotenv>=0.5.2",
+    "setuptools>=82.0.1",
     "twine>=6.2.0",
 ]
 

confamnode-0.1.1/tests/test_ansa.py

@@ -0,0 +1,231 @@
+import pytest
+from confamnode.ansa import Ansa, Usage, Cost
+
+
+def test_usage_has_prompt_tokens():
+    usage = Usage(prompt_tokens=10, completion_tokens=20, total_tokens=30)
+    assert usage.prompt_tokens == 10
+
+
+def test_usage_has_completion_tokens():
+    usage = Usage(prompt_tokens=10, completion_tokens=20, total_tokens=30)
+    assert usage.completion_tokens == 20
+
+
+def test_usage_has_total_tokens():
+    usage = Usage(prompt_tokens=10, completion_tokens=20, total_tokens=30)
+    assert usage.total_tokens == 30
+
+
+def test_cost_has_naira():
+    cost = Cost(naira=1.23)
+    assert cost.naira == 1.23
+
+
+def test_cost_dollars_is_none_by_default():
+    cost = Cost(naira=1.23)
+    assert cost.dollars is None
+
+
+def test_cost_accepts_dollars():
+    cost = Cost(naira=1.23, dollars=0.0008)
+    assert cost.dollars == 0.0008
+
+
+def test_ansa_has_text():
+    usage = Usage(prompt_tokens=10, completion_tokens=20, total_tokens=30)
+    cost = Cost(naira=1.23)
+    ansa = Ansa(
+        text="How far!",
+        model="confam-speed",
+        usage=usage,
+        cost=cost,
+        finish_reason="stop",
+        raw={}
+    )
+    assert ansa.text == "How far!"
+
+
+def test_ansa_has_model():
+    usage = Usage(prompt_tokens=10, completion_tokens=20, total_tokens=30)
+    cost = Cost(naira=1.23)
+    ansa = Ansa(
+        text="How far!",
+        model="confam-speed",
+        usage=usage,
+        cost=cost,
+        finish_reason="stop",
+        raw={}
+    )
+    assert ansa.model == "confam-speed"
+
+
+def test_ansa_reasoning_is_none_by_default():
+    usage = Usage(prompt_tokens=10, completion_tokens=20, total_tokens=30)
+    cost = Cost(naira=1.23)
+    ansa = Ansa(
+        text="How far!",
+        model="confam-speed",
+        usage=usage,
+        cost=cost,
+        finish_reason="stop",
+        raw={}
+    )
+    assert ansa.reasoning is None
+
+
+def test_ansa_tools_is_empty_list_by_default():
+    usage = Usage(prompt_tokens=10, completion_tokens=20, total_tokens=30)
+    cost = Cost(naira=1.23)
+    ansa = Ansa(
+        text="How far!",
+        model="confam-speed",
+        usage=usage,
+        cost=cost,
+        finish_reason="stop",
+        raw={}
+    )
+    assert ansa.tools == []
+
+
+def test_ansa_has_usage():
+    usage = Usage(prompt_tokens=10, completion_tokens=20, total_tokens=30)
+    cost = Cost(naira=1.23)
+    ansa = Ansa(
+        text="How far!",
+        model="confam-speed",
+        usage=usage,
+        cost=cost,
+        finish_reason="stop",
+        raw={}
+    )
+    assert ansa.usage.total_tokens == 30
+
+
+def test_ansa_has_finish_reason():
+    usage = Usage(prompt_tokens=10, completion_tokens=20, total_tokens=30)
+    cost = Cost(naira=1.23)
+    ansa = Ansa(
+        text="How far!",
+        model="confam-speed",
+        usage=usage,
+        cost=cost,
+        finish_reason="stop",
+        raw={}
+    )
+    assert ansa.finish_reason == "stop"
+
+
+def test_ansa_has_raw():
+    usage = Usage(prompt_tokens=10, completion_tokens=20, total_tokens=30)
+    cost = Cost(naira=1.23)
+    raw = {"id": "abc123"}
+    ansa = Ansa(
+        text="How far!",
+        model="confam-speed",
+        usage=usage,
+        cost=cost,
+        finish_reason="stop",
+        raw=raw
+    )
+    assert ansa.raw == {"id": "abc123"}
+
+
+def test_ansa_accepts_reasoning():
+    usage = Usage(prompt_tokens=10, completion_tokens=20, total_tokens=30)
+    cost = Cost(naira=1.23)
+    ansa = Ansa(
+        text="The answer is 42",
+        model="confam-reasoning",
+        usage=usage,
+        cost=cost,
+        finish_reason="stop",
+        raw={},
+        reasoning="First I thought about it..."
+    )
+    assert ansa.reasoning == "First I thought about it..."
+
+
+def test_ansa_accepts_tools():
+    usage = Usage(prompt_tokens=10, completion_tokens=20, total_tokens=30)
+    cost = Cost(naira=1.23)
+    ansa = Ansa(
+        text="",
+        model="confam-agents",
+        usage=usage,
+        cost=cost,
+        finish_reason="tool_calls",
+        raw={},
+        tools=[{"name": "search", "arguments": {"query": "Lagos weather"}}]
+    )
+    assert len(ansa.tools) == 1
+    assert ansa.tools[0]["name"] == "search"
+
+
+def test_ansa_has_id():
+    usage = Usage(prompt_tokens=10, completion_tokens=20, total_tokens=30)
+    cost = Cost(naira=1.23)
+    ansa = Ansa(
+        text="How far!",
+        model="confam-speed",
+        usage=usage,
+        cost=cost,
+        finish_reason="stop",
+        raw={},
+    )
+    assert ansa.id.startswith("confam-")
+
+
+def test_ansa_id_is_unique():
+    usage = Usage(prompt_tokens=10, completion_tokens=20, total_tokens=30)
+    cost = Cost(naira=1.23)
+    ansa1 = Ansa(
+        text="How far!",
+        model="confam-speed",
+        usage=usage,
+        cost=cost,
+        finish_reason="stop",
+        raw={},
+    )
+    ansa2 = Ansa(
+        text="How far!",
+        model="confam-speed",
+        usage=usage,
+        cost=cost,
+        finish_reason="stop",
+        raw={},
+    )
+    assert ansa1.id != ansa2.id
+
+
+def test_ansa_citations_is_empty_list_by_default():
+    usage = Usage(prompt_tokens=10, completion_tokens=20, total_tokens=30)
+    cost = Cost(naira=1.23)
+    ansa = Ansa(
+        text="How far!",
+        model="confam-speed",
+        usage=usage,
+        cost=cost,
+        finish_reason="stop",
+        raw={}
+    )
+    assert ansa.citations == []
+
+
+def test_ansa_accepts_citations():
+    usage = Usage(prompt_tokens=10, completion_tokens=20, total_tokens=30)
+    cost = Cost(naira=1.23)
+    ansa = Ansa(
+        text="Nigerian GDP grew by 3.4%",
+        model="confam-speed",
+        usage=usage,
+        cost=cost,
+        finish_reason="stop",
+        raw={},
+        citations=[
+            {"source": "World Bank", "url": "https://worldbank.org/"},
+            {"source": "NBS Nigeria", "url": "https://nigerianstat.gov.ng/"}
+        ]
+    )
+    assert len(ansa.citations) == 2
+    assert ansa.citations[0]["source"] == "World Bank"

{confamnode-0.1.0 → confamnode-0.1.1}/tests/test_client.py

@@ -1,6 +1,7 @@
 import os
 import pytest
 from unittest.mock import patch
+
 from confamnode.client import ConfamNode
 from confamnode.exceptions import ConfamAuthError
 

{confamnode-0.1.0 → confamnode-0.1.1}/tests/test_gist.py

@@ -1,10 +1,13 @@
 import pytest
 from unittest.mock import patch, MagicMock
-from confamnode.client import ConfamNode
+
 from confamnode import models
+from confamnode.client import ConfamNode
+from confamnode.ansa import Ansa, Usage, Cost
 from confamnode.exceptions import ConfamModelError
 
 
+
 @pytest.fixture
 def client():
     return ConfamNode(api_key="confam-sk-test-abc123")
@@ -41,11 +44,11 @@ def test_gist_rejects_invalid_model(client):
 
 def test_gist_returns_content(client, mock_litellm_response):
     with patch("confamnode.client.litellm.completion", return_value=mock_litellm_response):
-        response = client.gist(
+        ansa = client.gist(
             model=models.SPEED,
             messages=[{"role": "user", "content": "How far?"}]
         )
-        assert response.choices[0].message.content == "How far! I dey fine."
+        assert ansa.text == "How far! I dey fine."
 
 
 def test_gist_accepts_all_valid_models(client, mock_litellm_response):
@@ -57,11 +60,11 @@ def test_gist_accepts_all_valid_models(client, mock_litellm_response):
     ]
     with patch("confamnode.client.litellm.completion", return_value=mock_litellm_response):
         for model in valid_models:
-            response = client.gist(
+            ansa = client.gist(
                 model=model,
                 messages=[{"role": "user", "content": "How far?"}]
             )
-            assert response is not None
+            assert ansa is not None
 
 
 def test_gist_sends_correct_model_to_litellm(client, mock_litellm_response):
@@ -77,11 +80,11 @@ def test_gist_sends_correct_model_to_litellm(client, mock_litellm_response):
 
 def test_gist_accepts_string_messages(client, mock_litellm_response):
     with patch("confamnode.client.litellm.completion", return_value=mock_litellm_response):
-        response = client.gist(
+        ansa = client.gist(
             model=models.SPEED,
             messages="How far?"
         )
-        assert response is not None
+        assert ansa is not None
 
 
 def test_gist_converts_string_to_message_list(client, mock_litellm_response):
@@ -96,11 +99,11 @@ def test_gist_converts_string_to_message_list(client, mock_litellm_response):
 
 def test_gist_accepts_lists_messages(client, mock_litellm_response):
     with patch("confamnode.client.litellm.completion", return_value=mock_litellm_response):
-        response = client.gist(
+        ansa = client.gist(
             model=models.SPEED,
             messages=[{"role": "user", "content": "How far?"}]
         )
-        assert response is not None
+        assert ansa is not None
 
 
 def test_gist_rejects_invalid_messages_type(client):
@@ -122,3 +125,91 @@ def test_gist_passes_kwargs_to_litellm(client, mock_litellm_response):
         call_args = mock_call.call_args
         assert call_args.kwargs["temperature"] == 0.7
         assert call_args.kwargs["max_tokens"] == 500
+
+
+def test_gist_prepends_openai_prefix_to_model(client, mock_litellm_response):
+    with patch("confamnode.client.litellm.completion", return_value=mock_litellm_response) as mock_call:
+        client.gist(
+            model="confam-speed",
+            messages="How far?"
+        )
+        call_args = mock_call.call_args
+        assert call_args.kwargs["model"] == "openai/confam-speed"
+
+
+def test_gist_returns_ansa_object(client, mock_litellm_response):
+    with patch("confamnode.client.litellm.completion", return_value=mock_litellm_response):
+        ansa = client.gist(
+            model="confam-speed",
+            messages="How far?"
+        )
+        assert isinstance(ansa, Ansa)
+
+
+def test_gist_has_test(client, mock_litellm_response):
+    with patch("confamnode.client.litellm.completion", return_value=mock_litellm_response):
+        ansa = client.gist(
+            model="confam-speed",
+            messages="How far?"
+        )
+        assert ansa.text == "How far! I dey fine."
+
+    
+def test_gist_has_model(client, mock_litellm_response):
+    with patch("confamnode.client.litellm.completion", return_value=mock_litellm_response):
+        ansa = client.gist(
+            model="confam-speed",
+            messages="How far?"
+        )
+        assert ansa.model == "confam-speed"
+
+
+def test_gist_has_usage(client, mock_litellm_response):
+    with patch("confamnode.client.litellm.completion", return_value=mock_litellm_response):
+        ansa = client.gist(
+            model="confam-speed",
+            messages="How far?"
+        )
+        assert isinstance(ansa.usage, Usage)
+        assert ansa.usage.prompt_tokens == 10
+        assert ansa.usage.completion_tokens == 20
+        assert ansa.usage.total_tokens == 30
+
+
+def test_gist_has_cost(client, mock_litellm_response):
+    with patch("confamnode.client.litellm.completion", return_value=mock_litellm_response):
+        ansa = client.gist(
+            model="confam-speed",
+            messages="How far?"
+        )
+        assert isinstance(ansa.cost, Cost)
+        assert isinstance(ansa.cost.naira, float)
+
+
+def test_gist_has_finish_reason(client, mock_litellm_response):
+    with patch("confamnode.client.litellm.completion", return_value=mock_litellm_response):
+        ansa = client.gist(
+            model="confam-speed",
+            messages="How far?"
+        )
+        assert ansa.finish_reason is not None
+
+
+def test_gist_has_id(client, mock_litellm_response):
+    with patch("confamnode.client.litellm.completion", return_value=mock_litellm_response):
+        ansa = client.gist(
+            model="confam-speed",
+            messages="How far?"
+        )
+        assert ansa.id.startswith("confam-")
+
+
+def test_gist_has_raw(client, mock_litellm_response):
+    with patch("confamnode.client.litellm.completion", return_value=mock_litellm_response):
+        ansa = client.gist(
+            model="confam-speed",
+            messages="How far?"
+        )
+        assert ansa.raw is not None
+
+

{confamnode-0.1.0 → confamnode-0.1.1}/tests/test_init.py

@@ -19,4 +19,9 @@ def test_exceptions_importable_from_package():
 def test_version_is_defined():
     import confamnode
     assert hasattr(confamnode, "__version__")
-    assert isinstance(confamnode.__version__, str)
+    assert isinstance(confamnode.__version__, str)
+
+
+def test_confamstream_importable_from_package():
+    from confamnode import ConfamStream
+    assert ConfamStream is not None

confamnode-0.1.1/tests/test_models.py

@@ -0,0 +1,84 @@
+from confamnode.models import (
+    LITE, SPEED, REASONING,
+    INTELLIGENCE, DEEP_REASONING, CODE, CODE_PRO, VISION, AUDIO, TTS,
+    NANO,
+    EMBED_TEXT, EMBED_MULTIMODAL, EMBED_MULTIMODAL_2, EMBED_TEXT_LOCAL,
+    EMBED_TEXT_PRO, EMBED_MULTIMODAL_PRO, EMBED_MULTILINGUAL,
+    EMBED_SMALL, EMBED_0_6B,
+    RERANK, RERANK_FAST,
+)
+
+
+def test_model_constants_are_strings():
+    assert isinstance(LITE, str)
+    assert isinstance(SPEED, str)
+    assert isinstance(REASONING, str)
+    assert isinstance(INTELLIGENCE, str)
+    assert isinstance(DEEP_REASONING, str)
+    assert isinstance(CODE, str)
+    assert isinstance(CODE_PRO, str)
+    assert isinstance(VISION, str)
+    assert isinstance(AUDIO, str)
+    assert isinstance(TTS, str)
+    assert isinstance(NANO, str)
+    assert isinstance(EMBED_TEXT, str)
+    assert isinstance(EMBED_MULTIMODAL, str)
+    assert isinstance(EMBED_MULTIMODAL_2, str)
+    assert isinstance(EMBED_TEXT_LOCAL, str)
+    assert isinstance(EMBED_TEXT_PRO, str)
+    assert isinstance(EMBED_MULTIMODAL_PRO, str)
+    assert isinstance(EMBED_MULTILINGUAL, str)
+    assert isinstance(EMBED_SMALL, str)
+    assert isinstance(EMBED_0_6B, str)
+    assert isinstance(RERANK, str)
+    assert isinstance(RERANK_FAST, str)
+
+
+def test_model_constants_start_with_confam():
+    assert LITE.startswith("confam-")
+    assert SPEED.startswith("confam-")
+    assert REASONING.startswith("confam-")
+    assert INTELLIGENCE.startswith("confam-")
+    assert DEEP_REASONING.startswith("confam-")
+    assert CODE.startswith("confam-")
+    assert CODE_PRO.startswith("confam-")
+    assert VISION.startswith("confam-")
+    assert AUDIO.startswith("confam-")
+    assert TTS.startswith("confam-")
+    assert NANO.startswith("confam-")
+    assert EMBED_TEXT.startswith("confam-")
+    assert EMBED_MULTIMODAL.startswith("confam-")
+    assert EMBED_MULTIMODAL_2.startswith("confam-")
+    assert EMBED_TEXT_LOCAL.startswith("confam-")
+    assert EMBED_TEXT_PRO.startswith("confam-")
+    assert EMBED_MULTIMODAL_PRO.startswith("confam-")
+    assert EMBED_MULTILINGUAL.startswith("confam-")
+    assert EMBED_SMALL.startswith("confam-")
+    assert EMBED_0_6B.startswith("confam-")
+    assert RERANK.startswith("confam-")
+    assert RERANK_FAST.startswith("confam-")
+
+
+def test_model_constants_have_correct_values():
+    assert LITE == "confam-lite"
+    assert SPEED == "confam-speed"
+    assert REASONING == "confam-reasoning"
+    assert INTELLIGENCE == "confam-intelligence"
+    assert DEEP_REASONING == "confam-deep-reasoning"
+    assert CODE == "confam-code"
+    assert CODE_PRO == "confam-code-pro"
+    assert VISION == "confam-vision"
+    assert AUDIO == "confam-audio"
+    assert TTS == "confam-tts"
+    assert NANO == "confam-nano"
+    assert EMBED_TEXT == "confam-embed-text-1"
+    assert EMBED_MULTIMODAL == "confam-embed-multimodal-1"
+    assert EMBED_MULTIMODAL_2 == "confam-embed-multimodal-2"
+    assert EMBED_TEXT_LOCAL == "confam-embed-text-local"
+    assert EMBED_TEXT_PRO == "confam-embed-text-pro"
+    assert EMBED_MULTIMODAL_PRO == "confam-embed-multimodal-pro"
+    assert EMBED_MULTILINGUAL == "confam-embed-multilingual"
+    assert EMBED_SMALL == "confam-embed-small"
+    assert EMBED_0_6B == "confam-embed-0.6b"
+    assert RERANK == "confam-rerank"
+    assert RERANK_FAST == "confam-rerank-fast"

{confamnode-0.1.0 → confamnode-0.1.1}/uv.lock

@@ -418,7 +418,7 @@ wheels = [
 
 [[package]]
 name = "confamnode"
-version = "0.1.0"
+version = "0.1.1"
 source = { editable = "." }
 dependencies = [
     { name = "litellm" },
@@ -430,6 +430,7 @@ dev = [
     { name = "build" },
     { name = "pytest" },
     { name = "pytest-dotenv" },
+    { name = "setuptools" },
     { name = "twine" },
 ]
 
@@ -444,6 +445,7 @@ dev = [
     { name = "build", specifier = ">=1.5.0" },
     { name = "pytest", specifier = ">=9.0.3" },
     { name = "pytest-dotenv", specifier = ">=0.5.2" },
+    { name = "setuptools", specifier = ">=82.0.1" },
     { name = "twine", specifier = ">=6.2.0" },
 ]
 
@@ -2283,6 +2285,15 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/b7/46/f5af3402b579fd5e11573ce652019a67074317e18c1935cc0b4ba9b35552/secretstorage-3.5.0-py3-none-any.whl", hash = "sha256:0ce65888c0725fcb2c5bc0fdb8e5438eece02c523557ea40ce0703c266248137", size = 15554, upload-time = "2025-11-23T19:02:51.545Z" },
 ]
 
+[[package]]
+name = "setuptools"
+version = "82.0.1"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/4f/db/cfac1baf10650ab4d1c111714410d2fbb77ac5a616db26775db562c8fab2/setuptools-82.0.1.tar.gz", hash = "sha256:7d872682c5d01cfde07da7bccc7b65469d3dca203318515ada1de5eda35efbf9", size = 1152316, upload-time = "2026-03-09T12:47:17.221Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/9d/76/f789f7a86709c6b087c5a2f52f911838cad707cc613162401badc665acfe/setuptools-82.0.1-py3-none-any.whl", hash = "sha256:a59e362652f08dcd477c78bb6e7bd9d80a7995bc73ce773050228a348ce2e5bb", size = 1006223, upload-time = "2026-03-09T12:47:15.026Z" },
+]
+
 [[package]]
 name = "shellingham"
 version = "1.5.4"

confamnode-0.1.0/confamnode/client.py

@@ -1,63 +0,0 @@
-import os
-import litellm
-
-from typing import Union, List, Dict
-
-from confamnode import models
-from confamnode.exceptions import ConfamAuthError, ConfamModelError
-
-VALID_MODELS = [
-    models.LITE,
-    models.SPEED,
-    models.REASONING,
-    models.VISION,
-    models.AUDIO,
-    models.TTS,
-    models.NANO
-]
-
-DEFAULT_BASE_URL = "https://api.confamnode.com/v1"
-
-
-class ConfamNode:
-    def __init__(
-        self,
-        api_key: str = None,
-        base_url: str = None
-    ):
-        # Pick up from environment if not provided
-        api_key = api_key or os.environ.get("CONFAMNODE_API_KEY")
-
-        if not api_key:
-            raise ValueError("api_key is required")
-        
-        if not api_key.startswith("confam-sk-"):
-            raise ConfamAuthError()
-        
-        self.api_key = api_key
-        self.litellm_key = api_key.removeprefix("confam-")
-        self.base_url = base_url or DEFAULT_BASE_URL
-
-    def gist(
-        self,
-        model: str,
-        messages: Union[str, List[Dict[str, str]]],
-        **kwargs
-    ) -> object:
-        if model not in VALID_MODELS:
-            raise ConfamModelError(model)
-        
-        if isinstance(messages, str):
-            messages = [{"role": "user", "content": messages}]
-        elif not isinstance(messages, list):
-            raise ValueError("messages must be a string or list")
-        
-        response = litellm.completion(
-            model=model,
-            messages=messages,
-            api_key=self.litellm_key,
-            base_url=self.base_url,
-            **kwargs
-        )
-
-        return response

confamnode-0.1.0/confamnode/models.py

@@ -1,22 +0,0 @@
-# Chat models
-LITE = "confam-lite"
-SPEED = "confam-speed"
-REASONING = "confam-reasoning"
-VISION = "confam-vision"
-AUDIO = "confam-audio"
-TTS = "confam-tts"
-
-
-# Embedding models
-EMBED_TEXT = "confam-embed-text-1"
-EMBED_MULTIMODAL = "confam-embed-multimodal-1"
-EMBED_MULTIMODAL_2 = "confam-embed-multimodal-2"
-
-
-# Local models
-NANO = "confam-nano"
-
-
-# Local Embedding models
-EMBED_TEXT_LOCAL = "confam-embed-text-local"
-

confamnode-0.1.0/main.py

@@ -1,6 +0,0 @@
-def main():
-    print("Hello from confamnode-sdk!")
-
-
-if __name__ == "__main__":
-    main()

confamnode-0.1.0/tests/test_models.py

@@ -1,55 +0,0 @@
-from confamnode.models import (
-    LITE,
-    SPEED,
-    REASONING,
-    VISION,
-    AUDIO,
-    TTS,
-    NANO,
-    EMBED_TEXT,
-    EMBED_MULTIMODAL,
-    EMBED_MULTIMODAL_2,
-    EMBED_TEXT_LOCAL
-)
-
-def test_model_constants_are_strings():
-    assert isinstance(LITE, str)
-    assert isinstance(SPEED, str)
-    assert isinstance(REASONING, str)
-    assert isinstance(VISION, str)
-    assert isinstance(AUDIO, str)
-    assert isinstance(TTS, str)
-    assert isinstance(NANO, str)
-    assert isinstance(EMBED_TEXT, str)
-    assert isinstance(EMBED_MULTIMODAL, str)
-    assert isinstance(EMBED_MULTIMODAL_2, str)
-    assert isinstance(EMBED_TEXT_LOCAL, str)
-
-
-def test_model_constants_start_with_confam():
-    assert LITE.startswith("confam-")
-    assert SPEED.startswith("confam-")
-    assert REASONING.startswith("confam-")
-    assert VISION.startswith("confam-")
-    assert AUDIO.startswith("confam-")
-    assert TTS.startswith("confam-")
-    assert NANO.startswith("confam-")
-    assert EMBED_TEXT.startswith("confam-")
-    assert EMBED_MULTIMODAL.startswith("confam-")
-    assert EMBED_MULTIMODAL_2.startswith("confam-")
-    assert EMBED_TEXT_LOCAL.startswith("confam-")
-
-
-def test_model_constants_have_correct_values():
-    assert LITE == "confam-lite"
-    assert SPEED == "confam-speed"
-    assert REASONING == "confam-reasoning"
-    assert VISION == "confam-vision"
-    assert AUDIO == "confam-audio"
-    assert TTS == "confam-tts"
-    assert NANO == "confam-nano"
-    assert EMBED_TEXT == "confam-embed-text-1"
-    assert EMBED_MULTIMODAL == "confam-embed-multimodal-1"
-    assert EMBED_MULTIMODAL_2 == "confam-embed-multimodal-2"
-    assert EMBED_TEXT_LOCAL == "confam-embed-text-local"
-

confamnode-0.1.0/tests/test_stream.py

@@ -1,63 +0,0 @@
-import pytest
-from unittest.mock import patch, MagicMock
-from confamnode.client import ConfamNode
-from confamnode import models
-
-
-@pytest.fixture
-def client():
-    return ConfamNode(api_key="confam-sk-abc123")
-
-
-@pytest.fixture
-def mock_stream_response():
-    chunk1 = MagicMock()
-    chunk1.choices[0].delta.content = "How"
-    chunk2 = MagicMock()
-    chunk2.choices[0].delta.content = " far!"
-    chunk3 = MagicMock()
-    chunk3.choices[0].delta.content = None
-    return iter([chunk1, chunk2, chunk3])
-
-
-def test_gist_accepts_stream_true(client, mock_stream_response):
-    with patch("confamnode.client.litellm.completion", return_value=mock_stream_response):
-        response = client.gist(
-            model=models.SPEED,
-            messages="How far?",
-            stream=True
-        )
-        assert response is not None
-
-
-def test_gist_stream_returns_chunks(client, mock_stream_response):
-    with patch("confamnode.client.litellm.completion", return_value=mock_stream_response):
-        chunks = list(client.gist(
-            model=models.SPEED,
-            messages="How far?",
-            stream=True
-        ))
-        assert len(chunks) == 3
-
-
-def test_gist_stream_chunks_have_content(client, mock_stream_response):
-    with patch("confamnode.client.litellm.completion", return_value=mock_stream_response):
-        chunks = list(client.gist(
-            model=models.SPEED,
-            messages="How far?",
-            stream=True
-        ))
-        assert chunks[0].choices[0].delta.content == "How"
-        assert chunks[1].choices[0].delta.content == " far!"
-
-
-def test_gist_stream_passes_stream_true_to_litellm(client, mock_stream_response):
-    with patch("confamnode.client.litellm.completion", return_value=mock_stream_response) as mock_call:
-        client.gist(
-            model=models.SPEED,
-            messages="How far?",
-            stream=True
-        )
-        call_args = mock_call.call_args
-        assert call_args.kwargs.get("stream") == True
-