lionagi 0.11.0__py3-none-any.whl → 0.12.1__py3-none-any.whl

lionagi/service/providers/exa_/models.py

@@ -1,160 +1,3 @@
-from enum import Enum
+from khive.providers.exa_ import ExaSearchRequest
 
-from pydantic import BaseModel, Field
-
-
-class CategoryEnum(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 LivecrawlEnum(str, Enum):
-    never = "never"
-    fallback = "fallback"
-    always = "always"
-
-
-class SearchTypeEnum(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 | LivecrawlEnum = Field(
-        default=LivecrawlEnum.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 | CategoryEnum = Field(
-        default=None,
-        description="A data category to focus on, such as 'company', 'research paper', 'news', etc.",
-    )
-    type: None | SearchTypeEnum = 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.",
-    )
+__all__ = ("ExaSearchRequest",)

lionagi/service/providers/perplexity_/models.py

@@ -1,144 +1,5 @@
-from enum import Enum
-from typing import Any
+from khive.providers.perplexity_ import (
+    PerplexityChatRequest as PerplexityChatCompletionRequest,
+)
 
-from pydantic import BaseModel, Field, model_validator
-
-
-class PerplexityRole(str, Enum):
-    """Roles allowed in Perplexity's messages."""
-
-    system = "system"
-    user = "user"
-    assistant = "assistant"
-
-
-class PerplexityMessage(BaseModel):
-    """
-    A single message in the conversation.
-    `role` can be 'system', 'user', or 'assistant'.
-    `content` is the text for that conversation turn.
-    """
-
-    role: PerplexityRole = Field(
-        ...,
-        description="The role of the speaker. Must be system, user, or assistant.",
-    )
-    content: str = Field(..., description="The text content of this message.")
-
-
-class PerplexityChatCompletionRequest(BaseModel):
-    """
-    Represents the request body for Perplexity's Chat Completions endpoint.
-    Endpoint: POST https://api.perplexity.ai/chat/completions
-    """
-
-    model: str = Field(
-        "sonar",
-        description="The model name, e.g. 'sonar', (the only model available at the time when this request model was updated, check doc for latest info).",
-    )
-    messages: list[PerplexityMessage] = Field(
-        ..., description="A list of messages forming the conversation so far."
-    )
-
-    # Optional parameters
-    frequency_penalty: float | None = Field(
-        default=None,
-        gt=0,
-        description=(
-            "Multiplicative penalty > 0. Values > 1.0 penalize repeated tokens more strongly. "
-            "Value=1.0 means no penalty. Incompatible with presence_penalty."
-        ),
-    )
-    presence_penalty: float | None = Field(
-        default=None,
-        ge=-2.0,
-        le=2.0,
-        description=(
-            "Penalizes tokens that have appeared so far (range -2 to 2). "
-            "Positive values encourage talking about new topics. Incompatible with frequency_penalty."
-        ),
-    )
-    max_tokens: int | None = Field(
-        default=None,
-        description=(
-            "Maximum number of completion tokens. If omitted, model generates tokens until it "
-            "hits stop or context limit."
-        ),
-    )
-    return_images: bool | None = Field(
-        default=None,
-        description="If True, attempt to return images (closed beta feature).",
-    )
-    return_related_questions: bool | None = Field(
-        default=None,
-        description="If True, attempt to return related questions (closed beta feature).",
-    )
-    search_domain_filter: list[Any] | None = Field(
-        default=None,
-        description=(
-            "List of domains to limit or exclude in the online search. Example: ['example.com', '-twitter.com']. "
-            "Supports up to 3 entries. (Closed beta feature.)"
-        ),
-    )
-    search_recency_filter: str | None = Field(
-        default=None,
-        description=(
-            "Returns search results within a specified time interval: 'month', 'week', 'day', or 'hour'."
-        ),
-    )
-    stream: bool | None = Field(
-        default=None,
-        description=(
-            "If True, response is returned incrementally via Server-Sent Events (SSE)."
-        ),
-    )
-    temperature: float | None = Field(
-        default=None,
-        ge=0.0,
-        lt=2.0,
-        description=(
-            "Controls randomness of sampling, range [0, 2). Higher => more random. "
-            "Defaults to 0.2."
-        ),
-    )
-    top_k: int | None = Field(
-        default=None,
-        ge=0,
-        le=2048,
-        description=(
-            "Top-K filtering. 0 disables top-k filtering. If set, only the top K tokens are considered. "
-            "We recommend altering either top_k or top_p, but not both."
-        ),
-    )
-    top_p: float | None = Field(
-        default=None,
-        ge=0.0,
-        le=1.0,
-        description=(
-            "Nucleus sampling threshold. We recommend altering either top_k or top_p, but not both."
-        ),
-    )
-
-    @model_validator(mode="before")
-    def validate_penalties(cls, values):
-        """
-        Disallow using both frequency_penalty != 1.0 and presence_penalty != 0.0 at once,
-        since the docs say they're incompatible.
-        """
-        freq_pen = values.get("frequency_penalty", 1.0)
-        pres_pen = values.get("presence_penalty", 0.0)
-
-        # The doc states frequency_penalty is incompatible with presence_penalty.
-        # We'll enforce that if presence_penalty != 0, frequency_penalty must be 1.0
-        # or vice versa. Adjust logic as needed.
-        if pres_pen != 0.0 and freq_pen != 1.0:
-            raise ValueError(
-                "presence_penalty is incompatible with frequency_penalty. "
-                "Please use only one: either presence_penalty=0 with freq_pen !=1, "
-                "or presence_penalty!=0 with freq_pen=1."
-            )
-        return values
-
-    def to_dict(self) -> dict:
-        """Return a dict suitable for JSON serialization and sending to Perplexity API."""
-        return self.model_dump(exclude_none=True)
+__all__ = ("PerplexityChatCompletionRequest",)