nous-genai 0.1.0__py3-none-any.whl

nous/genai/reference/model_catalog_data/google.py

@@ -0,0 +1,45 @@
+from __future__ import annotations
+
+MODELS: list[str] = [
+    "aqa",
+    "deep-research-pro-preview-12-2025",
+    "gemini-2.0-flash",
+    "gemini-2.0-flash-lite",
+    "gemini-2.5-computer-use-preview-10-2025",
+    "gemini-2.5-flash",
+    "gemini-2.5-flash-lite",
+    "gemini-2.5-flash-native-audio-latest",
+    "gemini-2.5-pro",
+    "gemini-3-flash-preview",
+    "gemini-3-pro-preview",
+    "gemini-exp-1206",
+    "gemini-flash-latest",
+    "gemini-flash-lite-latest",
+    "gemini-pro-latest",
+    "gemini-robotics-er-1.5-preview",
+    "gemma-3-12b-it",
+    "gemma-3-1b-it",
+    "gemma-3-27b-it",
+    "gemma-3-4b-it",
+    "gemma-3n-e2b-it",
+    "gemma-3n-e4b-it",
+    "gemini-2.5-flash-image",
+    "gemini-3-pro-image-preview",
+    "imagen-4.0-fast-generate-001",
+    "imagen-4.0-generate-001",
+    "imagen-4.0-generate-preview-06-06",
+    "imagen-4.0-ultra-generate-001",
+    "veo-2.0-generate-001",
+    "veo-3.0-fast-generate-001",
+    "veo-3.0-generate-001",
+    "veo-3.1-fast-generate-preview",
+    "veo-3.1-generate-preview",
+    "gemini-2.5-flash-preview-tts",
+    "gemini-2.5-pro-preview-tts",
+    "embedding-001",
+    "embedding-gecko-001",
+    "gemini-embedding-001",
+    "gemini-embedding-exp",
+    "gemini-embedding-exp-03-07",
+    "text-embedding-004",
+]

nous/genai/reference/model_catalog_data/openai.py

@@ -0,0 +1,44 @@
+from __future__ import annotations
+
+MODELS: list[str] = [
+    "gpt-4.1",
+    "gpt-4.1-mini",
+    "gpt-4.1-nano",
+    "gpt-4o",
+    "gpt-4o-mini",
+    "chatgpt-4o-latest",
+    "gpt-5",
+    "gpt-5-codex",
+    "gpt-5-mini",
+    "gpt-5-nano",
+    "gpt-5-pro",
+    "gpt-5.1",
+    "gpt-5.1-codex",
+    "gpt-5.1-codex-max",
+    "gpt-5.1-codex-mini",
+    "gpt-5.2",
+    "gpt-5.2-pro",
+    "o1",
+    "o1-pro",
+    "o3",
+    "o3-mini",
+    "o4-mini",
+    "gpt-4o-mini-transcribe",
+    "gpt-4o-transcribe",
+    "gpt-4o-transcribe-diarize",
+    "whisper-1",
+    "dall-e-2",
+    "dall-e-3",
+    "gpt-image-1",
+    "gpt-image-1-mini",
+    "gpt-image-1.5",
+    "chatgpt-image-latest",
+    "sora-2",
+    "sora-2-pro",
+    "gpt-4o-mini-tts",
+    "tts-1",
+    "tts-1-hd",
+    "text-embedding-3-large",
+    "text-embedding-3-small",
+    "text-embedding-ada-002",
+]

nous/genai/reference/model_catalog_data/tuzi_anthropic.py

@@ -0,0 +1,21 @@
+from __future__ import annotations
+
+MODELS: list[str] = [
+    "claude-instant-1.2",
+    "claude-3-5-haiku-latest",
+    "claude-3-5-sonnet-latest",
+    "claude-3-7-sonnet-latest",
+    "claude-3-7-sonnet-latest-thinking",
+    "claude-sonnet-4-0",
+    "claude-sonnet-4-0-thinking",
+    "claude-opus-4-0",
+    "claude-opus-4-0-thinking",
+    "claude-opus-4-1",
+    "claude-haiku-4-5",
+    "claude-sonnet-4-5",
+    "claude-opus-4-5",
+    "Cursor-c4",
+    "Cursor-c4-thinking",
+    "Cursor-co4",
+    "Cursor-co4-thinking",
+]

nous/genai/reference/model_catalog_data/tuzi_google.py

@@ -0,0 +1,19 @@
+from __future__ import annotations
+
+MODELS: list[str] = [
+    "gemini-2.5-flash",
+    "gemini-2.5-flash-lite",
+    "gemini-2.5-flash-nothink",
+    "gemini-2.5-flash-search",
+    "gemini-2.5-pro",
+    "gemini-2.5-pro-search",
+    "gemini-3-flash-preview",
+    "gemini-3-pro-preview",
+    "gemini-3-pro-preview-search",
+    "gemini-2.0-flash-exp-image-generation",
+    "gemini-2.0-flash-preview-image-generation",
+    "gemini-2.5-flash-image",
+    "gemini-3-pro-image-preview",
+    "gemini-embedding-001",
+    "gemini-embedding-exp-03-07",
+]

nous/genai/reference/model_catalog_data/tuzi_openai.py

@@ -0,0 +1,75 @@
+from __future__ import annotations
+
+MODELS: list[str] = [
+    "computer-use-preview",
+    "gpt-3.5-turbo",
+    "gpt-3.5-turbo-16k",
+    "gpt-3.5-turbo-instruct",
+    "gpt-4",
+    "gpt-4-32k",
+    "gpt-4-turbo",
+    "gpt-4-vision-preview",
+    "gpt-4.1",
+    "gpt-4.1-mini",
+    "gpt-4.1-nano",
+    "gpt-4.5-preview",
+    "gpt-4o",
+    "gpt-4o-audio-preview",
+    "gpt-4o-mini",
+    "gpt-4o-mini-audio-preview",
+    "gpt-4o-mini-realtime-preview",
+    "gpt-4o-mini-search-preview",
+    "gpt-4o-realtime-preview",
+    "gpt-4o-search-preview",
+    "gpt-5",
+    "gpt-5-codex",
+    "gpt-5-mini",
+    "gpt-5-nano",
+    "gpt-5-pro",
+    "gpt-5.1",
+    "gpt-5.1-codex",
+    "gpt-5.1-codex-max",
+    "gpt-5.1-codex-mini",
+    "gpt-5.2",
+    "gpt-5.2-pro",
+    "gpt-oss-120b",
+    "GPT-OSS-20B",
+    "MiniMax-Text-01",
+    "o1",
+    "o1-mini",
+    "o1-pro",
+    "o3",
+    "o3-mini",
+    "o3-mini-high",
+    "o3-mini-low",
+    "o3-mini-medium",
+    "o3-pro",
+    "o4-mini",
+    "openai/gpt-oss-120b",
+    "openai/gpt-oss-20b",
+    "distil-whisper-large-v3-en",
+    "gpt-4o-mini-transcribe",
+    "gpt-4o-transcribe",
+    "gpt-4o-transcribe-diarize",
+    "whisper-1",
+    "whisper-large-v3",
+    "whisper-large-v3-turbo",
+    "dall-e-2",
+    "dall-e-3",
+    "FLUX.1-Kontext-pro",
+    "gpt-4o-image",
+    "gpt-image-1",
+    "gpt-image-1.5",
+    "sora-2",
+    "sora-2-pro",
+    "gpt-4o-mini-tts",
+    "tts-1",
+    "tts-1-hd",
+    "doubao-embedding-large-text-240915",
+    "doubao-embedding-text-240715",
+    "text-embedding-004",
+    "text-embedding-3-large",
+    "text-embedding-3-small",
+    "text-embedding-ada-002",
+    "text-embedding-v4",
+]

nous/genai/reference/model_catalog_data/tuzi_web.py

@@ -0,0 +1,136 @@
+from __future__ import annotations
+
+MODELS: list[str] = [
+    "claude-haiku-4-5-20251001",
+    "claude-haiku-4-5-20251001-thinking",
+    "claude-sonnet-4-5-20250929",
+    "claude-sonnet-4-5-20250929-thinking",
+    "claude-opus-4-5-20251101",
+    "claude-opus-4-5-20251101-thinking",
+    "deepseek-chat",
+    "deepseek-coder",
+    "deepseek-r1",
+    "deepseek-r1-all",
+    "deepseek-r1-search",
+    "deepseek-r1-search-pro",
+    "deepseek-r1-search-pro-thinking",
+    "deepseek-search",
+    "deepseek-v3",
+    "deepseek-v3.2",
+    "deepseek-v3.2-exp-thinking",
+    "deepseek-v3.2-thinking",
+    "doubao-seed-1-8-251228",
+    "doubao-seed-1-8-251228-thinking",
+    "gemini-2.5-flash",
+    "gemini-2.5-flash-deepsearch",
+    "gemini-2.5-pro",
+    "gemini-2.5-pro-deepsearch",
+    "gemini-3-flash-preview",
+    "gemini-3-flash-preview-all",
+    "gemini-3-flash-preview-thinking",
+    "gemini-3-pro",
+    "gemini-3-pro-deepsearch",
+    "gemini-3-pro-preview",
+    "gemini-3-pro-preview-thinking",
+    "gemini-3-pro-thinking",
+    "glm-4.5",
+    "glm-4.5-air",
+    "glm-4.5-airx",
+    "glm-4.5-flash",
+    "glm-4.5-x",
+    "glm-4v",
+    "gpt-4o",
+    "gpt-4o-mini",
+    "gpt-4o-search",
+    "gpt-4o-study",
+    "gpt-5",
+    "gpt-5-mini",
+    "gpt-5-nano",
+    "gpt-5-pro",
+    "gpt-5-thinking-all",
+    "gpt-5.1",
+    "gpt-5.1-thinking",
+    "gpt-5.1-thinking-all",
+    "gpt-5.2",
+    "gpt-oss-120b",
+    "gpt-oss-20b",
+    "grok-3",
+    "grok-3-deepersearch",
+    "grok-3-deepsearch",
+    "grok-3-reasoner",
+    "grok-3-think",
+    "grok-4",
+    "grok-4.1",
+    "kimi-k2-instruct",
+    "o1",
+    "o1-mini",
+    "o1-pro",
+    "o1-pro-all",
+    "o3",
+    "o3-all",
+    "o3-mini",
+    "o3-mini-high",
+    "o3-pro",
+    "o4-mini",
+    "qwen2.5-72b-instruct",
+    "qwen3-235b-a22b",
+    "qwen3-32b",
+    "search",
+    "search-gpts",
+    "search-gpts-chat",
+    "dall-e-3",
+    "doubao-seedream-4-0-250828",
+    "doubao-seedream-4-5-251128",
+    "flux-kontext-max",
+    "flux-kontext-pro",
+    "gemini-2.5-flash-image",
+    "gemini-3-pro-image-preview",
+    "gemini-3-pro-image-preview-2k",
+    "gemini-3-pro-image-preview-4k",
+    "gpt-image-1",
+    "gpt-image-1.5",
+    "kling_image",
+    "seededit",
+    "seedream-3.0",
+    "seedream-4-0-250828",
+    "seedream-v4",
+    "kling-video-o1",
+    "kling-video-o1-edit",
+    "kling_video",
+    "sora-16:9-480p-10s",
+    "sora-16:9-480p-5s",
+    "sora-16:9-720p-5s",
+    "sora-1:1-480p-10s",
+    "sora-1:1-480p-5s",
+    "sora-1:1-720p-5s",
+    "sora-2",
+    "sora-2-character",
+    "sora-2-pro",
+    "sora-2-pro-character",
+    "sora-9:16-480p-10",
+    "sora-9:16-480p-5s",
+    "sora-9:16-720p-5s",
+    "veo2",
+    "veo2-fast",
+    "veo2-fast-frames",
+    "veo2-pro",
+    "veo3",
+    "veo3-fast",
+    "veo3-fast-frames",
+    "veo3-frames",
+    "veo3-pro",
+    "veo3-pro-frames",
+    "veo3.1",
+    "veo3.1-components",
+    "veo3.1-pro",
+    "chirp-auk",
+    "chirp-bluejay",
+    "chirp-crow",
+    "chirp-v4",
+    "suno-v3",
+    "suno_lyrics",
+    "sonar-medium-chat",
+    "sonar-medium-online",
+    "sonar-small-chat",
+    "sonar-small-online",
+]

nous/genai/reference/model_catalog_data/volcengine.py

@@ -0,0 +1,107 @@
+from __future__ import annotations
+
+MODELS: list[str] = [
+    "deepseek-r1-250120",
+    "deepseek-r1-250528",
+    "deepseek-r1-distill-qwen-32b-250120",
+    "deepseek-r1-distill-qwen-7b-250120",
+    "deepseek-v3-1-250821",
+    "deepseek-v3-1-terminus",
+    "deepseek-v3-2-251201",
+    "deepseek-v3-241226",
+    "deepseek-v3-250324",
+    "doubao-1-5-lite-32k-250115",
+    "doubao-1-5-pro-256k-250115",
+    "doubao-1-5-pro-32k-250115",
+    "doubao-1-5-pro-32k-character-250228",
+    "doubao-1-5-pro-32k-character-250715",
+    "doubao-1-5-thinking-pro-250415",
+    "doubao-1-5-thinking-pro-m-250415",
+    "doubao-1-5-thinking-pro-m-250428",
+    "doubao-1-5-thinking-vision-pro-250428",
+    "doubao-1-5-ui-tars-250428",
+    "doubao-1-5-vision-pro-32k-250115",
+    "doubao-1.5-ui-tars-250328",
+    "doubao-1.5-vision-lite-250315",
+    "doubao-1.5-vision-pro-250328",
+    "doubao-embedding-vision-241215",
+    "doubao-embedding-vision-250328",
+    "doubao-embedding-vision-250615",
+    "doubao-embedding-vision-251215",
+    "doubao-lite-128k-240428",
+    "doubao-lite-128k-240828",
+    "doubao-lite-32k-240428",
+    "doubao-lite-32k-240628",
+    "doubao-lite-32k-240828",
+    "doubao-lite-32k-character-241015",
+    "doubao-lite-32k-character-250228",
+    "doubao-lite-4k-240328",
+    "doubao-lite-4k-character-240515",
+    "doubao-lite-4k-character-240828",
+    "doubao-lite-4k-pretrain-character-240516",
+    "doubao-pro-128k-240515",
+    "doubao-pro-128k-240628",
+    "doubao-pro-256k-241115",
+    "doubao-pro-32k-240615",
+    "doubao-pro-32k-240828",
+    "doubao-pro-32k-241215",
+    "doubao-pro-32k-browsing-240615",
+    "doubao-pro-32k-browsing-240828",
+    "doubao-pro-32k-browsing-241115",
+    "doubao-pro-32k-character-240528",
+    "doubao-pro-32k-character-240828",
+    "doubao-pro-32k-character-241215",
+    "doubao-pro-32k-functioncall-240515",
+    "doubao-pro-32k-functioncall-240815",
+    "doubao-pro-32k-functioncall-241028",
+    "doubao-pro-32k-functioncall-preview",
+    "doubao-pro-4k-240515",
+    "doubao-pro-4k-browsing-240524",
+    "doubao-pro-4k-character-240515",
+    "doubao-pro-4k-character-240728",
+    "doubao-pro-4k-functioncall-240515",
+    "doubao-pro-4k-functioncall-240615",
+    "doubao-seaweed-241128",
+    "doubao-seed-1-6-250615",
+    "doubao-seed-1-6-251015",
+    "doubao-seed-1-6-flash-250615",
+    "doubao-seed-1-6-flash-250715",
+    "doubao-seed-1-6-flash-250828",
+    "doubao-seed-1-6-lite-251015",
+    "doubao-seed-1-6-thinking-250615",
+    "doubao-seed-1-6-thinking-250715",
+    "doubao-seed-1-6-vision-250815",
+    "doubao-seed-1-8-251228",
+    "doubao-seed-code-preview-251028",
+    "doubao-seed-translation-250915",
+    "doubao-seed3d-1-0-250928",
+    "doubao-seedance-1-0-pro-250528",
+    "doubao-seedance-1-0-pro-fast-251015",
+    "doubao-seedance-1-5-pro-251215",
+    "doubao-seededit-3-0-i2i-250628",
+    "doubao-smart-router-250928",
+    "doubao-vision-lite-32k-241015",
+    "doubao-vision-pro-32k-241028",
+    "glm-4-5-air-20250728",
+    "kimi-k2-250711",
+    "kimi-k2-250905",
+    "kimi-k2-thinking-251104",
+    "mistral-7b-instruct-v0.2",
+    "qwen2-5-72b-20240919",
+    "qwen3-0-6b-20250429",
+    "qwen3-14b-20250429",
+    "qwen3-32b-20250429",
+    "qwen3-8b-20250429",
+    "wan2-1-14b-flf2v-250417",
+    "wan2-1-14b-i2v-250225",
+    "wan2-1-14b-t2v-250225",
+    "doubao-seedream-3-0-t2i-250415",
+    "doubao-seedream-4-0-250828",
+    "doubao-seedream-4-5-251128",
+    "doubao-seedance-1-0-lite-i2v-250428",
+    "doubao-seedance-1-0-lite-t2v-250428",
+    "doubao-embedding-large-text-240915",
+    "doubao-embedding-large-text-250515",
+    "doubao-embedding-text-240515",
+    "doubao-embedding-text-240715",
+]

nous/genai/tools/__init__.py

@@ -0,0 +1,13 @@
+from .output_parser import (
+    DEFAULT_OUTPUT_PARSER_TOOL_NAME,
+    build_output_parser_tool,
+    extract_output_from_response,
+    parse_output,
+)
+
+__all__ = [
+    "DEFAULT_OUTPUT_PARSER_TOOL_NAME",
+    "build_output_parser_tool",
+    "extract_output_from_response",
+    "parse_output",
+]

nous/genai/tools/output_parser.py

@@ -0,0 +1,119 @@
+from __future__ import annotations
+
+from typing import Any
+
+from .._internal.errors import invalid_request_error, not_supported_error
+from .._internal.json_schema import normalize_json_schema
+from ..client import Client
+from ..types import (
+    GenerateRequest,
+    GenerateResponse,
+    Message,
+    OutputSpec,
+    Part,
+    Tool,
+    ToolChoice,
+)
+
+DEFAULT_OUTPUT_PARSER_TOOL_NAME = "nous_output_parser"
+
+_DEFAULT_WRAPPER_KEY = "output"
+
+
+def build_output_parser_tool(
+    json_schema: Any,
+    *,
+    name: str = DEFAULT_OUTPUT_PARSER_TOOL_NAME,
+    description: str | None = None,
+) -> Tool:
+    """
+    Build a provider-agnostic function tool used for "structured outputs".
+
+    The model is expected to return the final answer by calling this tool with
+    a single argument key `"output"` matching the given JSON Schema.
+    """
+    tool_name = name.strip()
+    if not tool_name:
+        raise invalid_request_error("output parser tool name must be non-empty")
+    schema = normalize_json_schema(json_schema)
+    parameters = {
+        "type": "object",
+        "properties": {_DEFAULT_WRAPPER_KEY: schema},
+        "required": [_DEFAULT_WRAPPER_KEY],
+        "additionalProperties": False,
+    }
+    desc = description
+    if not isinstance(desc, str) or not desc.strip():
+        desc = (
+            "Return the final result by calling this tool with a single JSON argument "
+            f"`{_DEFAULT_WRAPPER_KEY}` matching the schema. Do not output extra text."
+        )
+    return Tool(name=tool_name, description=desc, parameters=parameters, strict=True)
+
+
+def extract_output_from_response(
+    response: GenerateResponse,
+    *,
+    tool_name: str = DEFAULT_OUTPUT_PARSER_TOOL_NAME,
+) -> Any:
+    """
+    Extract the structured output object from a provider response that contains
+    a tool_call part for `tool_name`.
+    """
+    name = tool_name.strip()
+    if not name:
+        raise invalid_request_error("tool_name must be non-empty")
+
+    for msg in response.output:
+        for part in msg.content:
+            if part.type != "tool_call":
+                continue
+            if part.meta.get("name") != name:
+                continue
+            arguments = part.meta.get("arguments")
+            if not isinstance(arguments, dict):
+                raise invalid_request_error(
+                    "output parser tool_call arguments must be an object"
+                )
+            if _DEFAULT_WRAPPER_KEY not in arguments:
+                raise invalid_request_error(
+                    f"output parser tool_call missing '{_DEFAULT_WRAPPER_KEY}'"
+                )
+            return arguments[_DEFAULT_WRAPPER_KEY]
+
+    raise invalid_request_error(f"missing output parser tool_call: {name}")
+
+
+def parse_output(
+    client: Client,
+    *,
+    model: str,
+    text: str,
+    json_schema: Any,
+    tool_name: str = DEFAULT_OUTPUT_PARSER_TOOL_NAME,
+) -> Any:
+    """
+    Parse plain text into a structured object by forcing a single tool call.
+    """
+    if not isinstance(text, str) or not text.strip():
+        raise invalid_request_error("text must be a non-empty string")
+    cap = client.capabilities(model)
+    if not cap.supports_tools:
+        raise not_supported_error("this model does not support tools")
+
+    tool = build_output_parser_tool(json_schema, name=tool_name)
+    prompt = (
+        "Convert the following text into structured output by calling the tool. "
+        "Call the tool only; do not output any other text.\n\n" + text.strip()
+    )
+    req = GenerateRequest(
+        model=model,
+        input=[Message(role="user", content=[Part.from_text(prompt)])],
+        output=OutputSpec(modalities=["text"]),
+        tools=[tool],
+        tool_choice=ToolChoice(mode="tool", name=tool.name),
+    )
+    resp = client.generate(req)
+    if not isinstance(resp, GenerateResponse):
+        raise not_supported_error("streaming responses are not supported")
+    return extract_output_from_response(resp, tool_name=tool.name)