lionagi 0.12.2__py3-none-any.whl → 0.12.4__py3-none-any.whl

lionagi/service/third_party/anthropic_models.py

@@ -0,0 +1,159 @@
+# Copyright (c) 2025, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+"""Anthropic API models for request/response validation."""
+
+from __future__ import annotations
+
+from typing import Literal, Optional, Union
+
+from pydantic import BaseModel, Field, field_validator
+
+
+class TextContentBlock(BaseModel):
+    type: Literal["text"] = "text"
+    text: str
+    cache_control: dict | None = None
+
+
+class ImageSource(BaseModel):
+    type: Literal["base64"] = "base64"
+    media_type: Literal["image/jpeg", "image/png", "image/gif", "image/webp"]
+    data: str
+
+
+class ImageContentBlock(BaseModel):
+    type: Literal["image"] = "image"
+    source: ImageSource
+
+
+ContentBlock = Union[TextContentBlock, ImageContentBlock]
+
+
+class Message(BaseModel):
+    role: Literal["user", "assistant"]
+    content: str | list[str | ContentBlock]
+
+    @field_validator("content", mode="before")
+    def validate_content(cls, v):
+        """Convert string content to proper format."""
+        if isinstance(v, str):
+            return v
+        if isinstance(v, list):
+            # Ensure all items are either strings or proper content blocks
+            result = []
+            for item in v:
+                if isinstance(item, str):
+                    result.append({"type": "text", "text": item})
+                else:
+                    result.append(item)
+            return result
+        return v
+
+
+class ToolDefinition(BaseModel):
+    name: str = Field(
+        ..., min_length=1, max_length=64, pattern="^[a-zA-Z0-9_-]+$"
+    )
+    description: str | None = None
+    input_schema: dict
+
+
+class ToolChoice(BaseModel):
+    type: Literal["auto", "any", "tool"]
+    name: str | None = None
+
+
+class CreateMessageRequest(BaseModel):
+    """Request model for Anthropic messages API."""
+
+    model: str = Field(..., min_length=1, max_length=256)
+    messages: list[Message]
+    max_tokens: int = Field(..., ge=1)
+
+    # Optional fields
+    system: str | list[ContentBlock] | None = None
+    temperature: float | None = Field(None, ge=0, le=1)
+    top_p: float | None = Field(None, ge=0, le=1)
+    top_k: int | None = Field(None, ge=0)
+    stop_sequences: list[str] | None = None
+    stream: bool | None = False
+    metadata: dict | None = None
+    tools: list[ToolDefinition] | None = None
+    tool_choice: ToolChoice | dict | None = None
+
+    class Config:
+        extra = "forbid"
+
+
+class Usage(BaseModel):
+    """Token usage information."""
+
+    input_tokens: int
+    output_tokens: int
+
+
+class ContentBlockResponse(BaseModel):
+    """Response content block."""
+
+    type: Literal["text"]
+    text: str
+
+
+class CreateMessageResponse(BaseModel):
+    """Response model for Anthropic messages API."""
+
+    id: str
+    type: Literal["message"] = "message"
+    role: Literal["assistant"] = "assistant"
+    content: list[ContentBlockResponse]
+    model: str
+    stop_reason: None | (
+        Literal["end_turn", "max_tokens", "stop_sequence", "tool_use"]
+    ) = None
+    stop_sequence: str | None = None
+    usage: Usage
+
+
+# Streaming response models
+class MessageStartEvent(BaseModel):
+    type: Literal["message_start"] = "message_start"
+    message: CreateMessageResponse
+
+
+class ContentBlockStartEvent(BaseModel):
+    type: Literal["content_block_start"] = "content_block_start"
+    index: int
+    content_block: ContentBlockResponse
+
+
+class ContentBlockDeltaEvent(BaseModel):
+    type: Literal["content_block_delta"] = "content_block_delta"
+    index: int
+    delta: dict
+
+
+class ContentBlockStopEvent(BaseModel):
+    type: Literal["content_block_stop"] = "content_block_stop"
+    index: int
+
+
+class MessageDeltaEvent(BaseModel):
+    type: Literal["message_delta"] = "message_delta"
+    delta: dict
+    usage: Usage | None = None
+
+
+class MessageStopEvent(BaseModel):
+    type: Literal["message_stop"] = "message_stop"
+
+
+StreamEvent = Union[
+    MessageStartEvent,
+    ContentBlockStartEvent,
+    ContentBlockDeltaEvent,
+    ContentBlockStopEvent,
+    MessageDeltaEvent,
+    MessageStopEvent,
+]

lionagi/service/third_party/exa_models.py

@@ -0,0 +1,165 @@
+from enum import Enum
+
+from pydantic import BaseModel, Field
+
+
+class SearchCategory(str, Enum):
+    company = "company"
+    research_paper = "research paper"
+    news = "news"
+    pdf = "pdf"
+    github = "github"
+    tweet = "tweet"
+    personal_site = "personal site"
+    linkedin_profile = "linkedin profile"
+    financial_report = "financial report"
+
+
+class LivecrawlType(str, Enum):
+    never = "never"
+    fallback = "fallback"
+    always = "always"
+
+
+class SearchType(str, Enum):
+    keyword = "keyword"
+    neural = "neural"
+    auto = "auto"
+
+
+class ContentsText(BaseModel):
+    includeHtmlTags: bool | None = Field(
+        default=False,
+        description="Whether to include HTML tags in the text. Set to True if you want"
+        " to retain HTML structure for the LLM to interpret.",
+    )
+    maxCharacters: int | None = Field(
+        default=None,
+        description="The maximum number of characters to return from the webpage text",
+    )
+
+
+class ContentsHighlights(BaseModel):
+    highlightsPerUrl: int | None = Field(
+        default=1,
+        description="The number of highlight snippets you want per page.",
+    )
+    numSentences: int | None = Field(
+        default=5,
+        description="Number of sentences to return in each highlight snippet.",
+    )
+    query: None | str = Field(
+        default=None,
+        description="A specific query used to generate the highlight snippets.",
+    )
+
+
+class ContentsSummary(BaseModel):
+    query: None | str = Field(
+        default=None,
+        description="A specific query used to generate a summary of the webpage.",
+    )
+
+
+class ContentsExtras(BaseModel):
+    links: int | None = Field(
+        default=None, description="Number of links to return from each page."
+    )
+    imageLinks: int | None = Field(
+        default=None, description="Number of images to return for each result."
+    )
+
+
+class Contents(BaseModel):
+    text: None | ContentsText = Field(
+        default=None,
+        description="Return full or partial text for each page, with optional HTML "
+        "structure or size limit.",
+    )
+    highlights: None | ContentsHighlights = Field(
+        default=None, description="Return snippet highlights for each page."
+    )
+    summary: None | ContentsSummary = Field(
+        default=None, description="Return a short summary of each page."
+    )
+    livecrawl: None | LivecrawlType = Field(
+        default=LivecrawlType.never,
+        description="Livecrawling setting for each page. Options: never, fallback, always.",
+    )
+    livecrawlTimeout: int | None = Field(
+        default=10000,
+        description="Timeout in milliseconds for livecrawling. Default 10000.",
+    )
+    subpages: int | None = Field(
+        default=None,
+        description="Number of subpages to crawl within each URL.",
+    )
+    subpageTarget: None | str | list[str] = Field(
+        default=None,
+        description="A target subpage or multiple subpages (list) to crawl, e.g. 'cited papers'.",
+    )
+    extras: None | ContentsExtras = Field(
+        default=None,
+        description="Additional extras like links or images to return for each page.",
+    )
+
+
+class ExaSearchRequest(BaseModel):
+    query: str = Field(
+        ...,
+        description="The main query string describing what you're looking for.",
+    )
+    category: None | SearchCategory = Field(
+        default=None,
+        description="A data category to focus on, such as 'company', 'research paper', 'news', etc.",
+    )
+    type: None | SearchType = Field(
+        default=None,
+        description="The type of search to run. Can be 'auto', 'keyword', or 'neural'.",
+    )
+    useAutoprompt: None | bool = Field(
+        default=False,
+        description="If True, Exa auto-optimizes your query for best results (neural or auto search only).",
+    )
+    numResults: int | None = Field(
+        default=10, description="Number of results to return. Default is 10."
+    )
+    includeDomains: None | list[str] = Field(
+        default=None,
+        description="List of domains you want to include exclusively.",
+    )
+    excludeDomains: None | list[str] = Field(
+        default=None,
+        description="List of domains you do NOT want to see in the results.",
+    )
+    startCrawlDate: None | str = Field(
+        default=None,
+        description="Include results crawled after this ISO date (e.g., '2023-01-01T00:00:00.000Z').",
+    )
+    endCrawlDate: None | str = Field(
+        default=None,
+        description="Include results crawled before this ISO date.",
+    )
+    startPublishedDate: None | str = Field(
+        default=None,
+        description="Only return results published after this ISO date.",
+    )
+    endPublishedDate: None | str = Field(
+        default=None,
+        description="Only return results published before this ISO date.",
+    )
+    includeText: None | list[str] = Field(
+        default=None,
+        description="Strings that must appear in the webpage text. Only a single string up to "
+        "5 words is currently supported.",
+    )
+    excludeText: None | list[str] = Field(
+        default=None,
+        description="Strings that must NOT appear in the webpage text. Only a single string up to "
+        "5 words is currently supported.",
+    )
+    contents: None | Contents = Field(
+        default=None,
+        description="Dict defining the different ways you want to retrieve webpage contents, "
+        "including text, highlights, or summaries.",
+    )