lionagi 0.12.3__py3-none-any.whl → 0.12.5__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/{providers/exa_/models.py → third_party/exa_models.py}

@@ -3,7 +3,7 @@ from enum import Enum
 from pydantic import BaseModel, Field
 
 
-class CategoryEnum(str, Enum):
+class SearchCategory(str, Enum):
     company = "company"
     research_paper = "research paper"
     news = "news"
@@ -15,13 +15,13 @@ class CategoryEnum(str, Enum):
     financial_report = "financial report"
 
 
-class LivecrawlEnum(str, Enum):
+class LivecrawlType(str, Enum):
     never = "never"
     fallback = "fallback"
     always = "always"
 
 
-class SearchTypeEnum(str, Enum):
+class SearchType(str, Enum):
     keyword = "keyword"
     neural = "neural"
     auto = "auto"
@@ -30,11 +30,12 @@ class SearchTypeEnum(str, Enum):
 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.",
+        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.",
+        description="The maximum number of characters to return from the webpage text",
     )
 
 
@@ -72,7 +73,8 @@ class ContentsExtras(BaseModel):
 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.",
+        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."
@@ -80,8 +82,8 @@ class Contents(BaseModel):
     summary: None | ContentsSummary = Field(
         default=None, description="Return a short summary of each page."
     )
-    livecrawl: None | LivecrawlEnum = Field(
-        default=LivecrawlEnum.never,
+    livecrawl: None | LivecrawlType = Field(
+        default=LivecrawlType.never,
         description="Livecrawling setting for each page. Options: never, fallback, always.",
     )
     livecrawlTimeout: int | None = Field(
@@ -107,11 +109,11 @@ class ExaSearchRequest(BaseModel):
         ...,
         description="The main query string describing what you're looking for.",
     )
-    category: None | CategoryEnum = Field(
+    category: None | SearchCategory = Field(
         default=None,
         description="A data category to focus on, such as 'company', 'research paper', 'news', etc.",
     )
-    type: None | SearchTypeEnum = Field(
+    type: None | SearchType = Field(
         default=None,
         description="The type of search to run. Can be 'auto', 'keyword', or 'neural'.",
     )
@@ -148,13 +150,16 @@ class ExaSearchRequest(BaseModel):
     )
     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.",
+        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.",
+        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.",
+        description="Dict defining the different ways you want to retrieve webpage contents, "
+        "including text, highlights, or summaries.",
     )