poe 0.1.0__tar.gz

poe-0.1.0/.github/workflows/ci.yml

@@ -0,0 +1,30 @@
+name: CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  test:
+    name: Test Python ${{ matrix.python-version }}
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.9", "3.12", "3.13"]
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install -e ".[dev]"
+
+      - name: Run tests
+        run: pytest tests/ -v

poe-0.1.0/.github/workflows/pypi.yml

@@ -0,0 +1,80 @@
+name: Publish to PyPI
+
+on:
+  release:
+    types: [published]
+
+  # Allow manual trigger
+  workflow_dispatch:
+
+permissions:
+  contents: read
+  id-token: write  # Required for trusted publishing (OIDC)
+
+jobs:
+  test:
+    name: Run tests
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.9", "3.12", "3.13"]
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install -e ".[dev]"
+
+      - name: Run tests
+        run: pytest tests/ -v
+
+  build:
+    name: Build package
+    runs-on: ubuntu-latest
+    needs: test
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.13"
+
+      - name: Install build tools
+        run: python -m pip install --upgrade pip build
+
+      - name: Build sdist and wheel
+        run: python -m build
+
+      - name: Upload build artifacts
+        uses: actions/upload-artifact@v4
+        with:
+          name: dist
+          path: dist/
+
+  publish:
+    name: Publish to PyPI
+    runs-on: ubuntu-latest
+    needs: build
+    environment:
+      name: pypi
+      url: https://pypi.org/p/poe
+    steps:
+      - name: Download build artifacts
+        uses: actions/download-artifact@v4
+        with:
+          name: dist
+          path: dist/
+
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        # Uses trusted publishing (OIDC) — no API token needed.
+        # Trusted publisher already configured on PyPI for:
+        #   repo: poe-platform/poe-python
+        #   workflow: pypi.yml

poe-0.1.0/.gitignore

@@ -0,0 +1,17 @@
+__pycache__/
+*.py[cod]
+*$py.class
+*.egg-info/
+dist/
+build/
+*.egg
+.eggs/
+.pytest_cache/
+.mypy_cache/
+.venv/
+venv/
+env/
+*.so
+.DS_Store
+*.swp
+*.swo

poe-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Poe Platform
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

poe-0.1.0/PKG-INFO

@@ -0,0 +1,89 @@
+Metadata-Version: 2.4
+Name: poe
+Version: 0.1.0
+Summary: Python client for the Poe API (api.poe.com)
+Project-URL: Homepage, https://github.com/poe-platform/poe-python
+Project-URL: Repository, https://github.com/poe-platform/poe-python
+Project-URL: Documentation, https://developer.poe.com
+Project-URL: Issues, https://github.com/poe-platform/poe-python/issues
+Author-email: Poe Platform <support@poe.com>
+License-Expression: MIT
+License-File: LICENSE
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.9
+Requires-Dist: httpx>=0.25.0
+Provides-Extra: dev
+Requires-Dist: pytest-asyncio>=0.21; extra == 'dev'
+Requires-Dist: pytest>=7.0; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# poe
+
+Official Python client for the [Poe](https://poe.com) chat completions API.
+
+## Installation
+
+```bash
+pip install poe
+```
+
+## Quick Start
+
+```python
+from poe import PoeClient
+
+client = PoeClient(api_key="your-poe-api-key")
+
+# Simple chat
+response = client.chat("What is Python?", model="GPT-4o")
+print(response["choices"][0]["message"]["content"])
+```
+
+## Streaming
+
+```python
+for chunk in client.stream("Tell me a story", model="GPT-4o"):
+    print(chunk, end="", flush=True)
+```
+
+## Options
+
+```python
+client = PoeClient(
+    api_key="your-key",
+    timeout=120.0,       # request timeout in seconds
+)
+
+response = client.chat(
+    "Explain quantum computing",
+    model="GPT-4o",
+    temperature=0.7,
+    max_tokens=1024,
+    system_prompt="You are a helpful physics teacher.",
+)
+```
+
+## Context Manager
+
+```python
+with PoeClient(api_key="your-key") as client:
+    resp = client.chat("Hello!")
+    print(resp["choices"][0]["message"]["content"])
+```
+
+## API Key
+
+Get your API key from [poe.com/api_key](https://poe.com/api_key).
+
+## License
+
+MIT

poe-0.1.0/README.md

@@ -0,0 +1,61 @@
+# poe
+
+Official Python client for the [Poe](https://poe.com) chat completions API.
+
+## Installation
+
+```bash
+pip install poe
+```
+
+## Quick Start
+
+```python
+from poe import PoeClient
+
+client = PoeClient(api_key="your-poe-api-key")
+
+# Simple chat
+response = client.chat("What is Python?", model="GPT-4o")
+print(response["choices"][0]["message"]["content"])
+```
+
+## Streaming
+
+```python
+for chunk in client.stream("Tell me a story", model="GPT-4o"):
+    print(chunk, end="", flush=True)
+```
+
+## Options
+
+```python
+client = PoeClient(
+    api_key="your-key",
+    timeout=120.0,       # request timeout in seconds
+)
+
+response = client.chat(
+    "Explain quantum computing",
+    model="GPT-4o",
+    temperature=0.7,
+    max_tokens=1024,
+    system_prompt="You are a helpful physics teacher.",
+)
+```
+
+## Context Manager
+
+```python
+with PoeClient(api_key="your-key") as client:
+    resp = client.chat("Hello!")
+    print(resp["choices"][0]["message"]["content"])
+```
+
+## API Key
+
+Get your API key from [poe.com/api_key](https://poe.com/api_key).
+
+## License
+
+MIT

poe-0.1.0/pyproject.toml

@@ -0,0 +1,44 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "poe"
+version = "0.1.0"
+description = "Python client for the Poe API (api.poe.com)"
+readme = "README.md"
+license = "MIT"
+requires-python = ">=3.9"
+authors = [
+    { name = "Poe Platform", email = "support@poe.com" },
+]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+]
+dependencies = [
+    "httpx>=0.25.0",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=7.0",
+    "pytest-asyncio>=0.21",
+]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/poe"]
+
+[project.urls]
+Homepage = "https://github.com/poe-platform/poe-python"
+Repository = "https://github.com/poe-platform/poe-python"
+Documentation = "https://developer.poe.com"
+Issues = "https://github.com/poe-platform/poe-python/issues"

poe-0.1.0/src/poe/__init__.py

@@ -0,0 +1,6 @@
+"""Python client for the Poe API."""
+
+from poe.client import PoeClient
+
+__all__ = ["PoeClient"]
+__version__ = "0.1.0"

poe-0.1.0/src/poe/client.py

@@ -0,0 +1,150 @@
+"""Poe API client for chat completions."""
+
+from __future__ import annotations
+
+from typing import Any, Generator, Iterator
+
+import httpx
+
+BASE_URL = "https://api.poe.com/v1"
+
+
+class PoeClient:
+    """Client for the Poe chat completions API.
+
+    Usage::
+
+        from poe import PoeClient
+
+        client = PoeClient(api_key="your-api-key")
+
+        # Non-streaming
+        response = client.chat("What is Python?", model="GPT-4o")
+        print(response["choices"][0]["message"]["content"])
+
+        # Streaming
+        for chunk in client.stream("Tell me a story", model="GPT-4o"):
+            print(chunk, end="", flush=True)
+    """
+
+    def __init__(
+        self,
+        api_key: str,
+        *,
+        base_url: str = BASE_URL,
+        timeout: float = 60.0,
+    ) -> None:
+        self.api_key = api_key
+        self.base_url = base_url.rstrip("/")
+        self._http = httpx.Client(
+            base_url=self.base_url,
+            headers={
+                "Authorization": f"Bearer {api_key}",
+                "Content-Type": "application/json",
+            },
+            timeout=timeout,
+        )
+
+    # -- public API -----------------------------------------------------------
+
+    def chat(
+        self,
+        message: str,
+        *,
+        model: str = "GPT-4o",
+        temperature: float | None = None,
+        max_tokens: int | None = None,
+        system_prompt: str | None = None,
+    ) -> dict[str, Any]:
+        """Send a chat completion request and return the full response."""
+        body = self._build_body(
+            message,
+            model=model,
+            stream=False,
+            temperature=temperature,
+            max_tokens=max_tokens,
+            system_prompt=system_prompt,
+        )
+        resp = self._http.post("/chat/completions", json=body)
+        resp.raise_for_status()
+        return resp.json()
+
+    def stream(
+        self,
+        message: str,
+        *,
+        model: str = "GPT-4o",
+        temperature: float | None = None,
+        max_tokens: int | None = None,
+        system_prompt: str | None = None,
+    ) -> Iterator[str]:
+        """Stream a chat completion, yielding content deltas as strings."""
+        body = self._build_body(
+            message,
+            model=model,
+            stream=True,
+            temperature=temperature,
+            max_tokens=max_tokens,
+            system_prompt=system_prompt,
+        )
+        with self._http.stream("POST", "/chat/completions", json=body) as resp:
+            resp.raise_for_status()
+            yield from self._iter_sse(resp)
+
+    # -- internals ------------------------------------------------------------
+
+    @staticmethod
+    def _build_body(
+        message: str,
+        *,
+        model: str,
+        stream: bool,
+        temperature: float | None,
+        max_tokens: int | None,
+        system_prompt: str | None,
+    ) -> dict[str, Any]:
+        messages: list[dict[str, str]] = []
+        if system_prompt:
+            messages.append({"role": "system", "content": system_prompt})
+        messages.append({"role": "user", "content": message})
+
+        body: dict[str, Any] = {
+            "model": model,
+            "messages": messages,
+            "stream": stream,
+        }
+        if temperature is not None:
+            body["temperature"] = temperature
+        if max_tokens is not None:
+            body["max_tokens"] = max_tokens
+        return body
+
+    @staticmethod
+    def _iter_sse(resp: httpx.Response) -> Generator[str, None, None]:
+        """Parse an SSE stream and yield content deltas."""
+        import json as _json
+
+        for line in resp.iter_lines():
+            if not line or not line.startswith("data: "):
+                continue
+            payload = line[len("data: "):]
+            if payload.strip() == "[DONE]":
+                return
+            try:
+                chunk = _json.loads(payload)
+            except _json.JSONDecodeError:
+                continue
+            delta = chunk.get("choices", [{}])[0].get("delta", {})
+            content = delta.get("content")
+            if content:
+                yield content
+
+    def close(self) -> None:
+        """Close the underlying HTTP client."""
+        self._http.close()
+
+    def __enter__(self) -> "PoeClient":
+        return self
+
+    def __exit__(self, *exc: Any) -> None:
+        self.close()

poe-0.1.0/tests/test_client.py

@@ -0,0 +1,41 @@
+"""Basic unit tests for PoeClient."""
+
+from poe.client import PoeClient
+
+
+def test_build_body_minimal():
+    body = PoeClient._build_body("Hello", model="GPT-4o", stream=False, temperature=None, max_tokens=None, system_prompt=None)
+    assert body == {
+        "model": "GPT-4o",
+        "messages": [{"role": "user", "content": "Hello"}],
+        "stream": False,
+    }
+
+
+def test_build_body_with_all_options():
+    body = PoeClient._build_body(
+        "Hi",
+        model="Claude-3.5-Sonnet",
+        stream=True,
+        temperature=0.5,
+        max_tokens=100,
+        system_prompt="Be concise.",
+    )
+    assert body["model"] == "Claude-3.5-Sonnet"
+    assert body["stream"] is True
+    assert body["temperature"] == 0.5
+    assert body["max_tokens"] == 100
+    assert body["messages"][0] == {"role": "system", "content": "Be concise."}
+    assert body["messages"][1] == {"role": "user", "content": "Hi"}
+
+
+def test_client_context_manager():
+    client = PoeClient(api_key="test-key")
+    with client:
+        assert client.api_key == "test-key"
+
+
+def test_client_default_base_url():
+    client = PoeClient(api_key="k")
+    assert client.base_url == "https://api.poe.com/v1"
+    client.close()