perplexityai 0.22.3__py3-none-any.whl

perplexity/types/shared/choice.py

@@ -0,0 +1,19 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from typing_extensions import Literal
+
+from ..._models import BaseModel
+from .chat_message_output import ChatMessageOutput
+
+__all__ = ["Choice"]
+
+
+class Choice(BaseModel):
+    delta: ChatMessageOutput
+
+    index: int
+
+    message: ChatMessageOutput
+
+    finish_reason: Optional[Literal["stop", "length"]] = None

perplexity/types/shared/usage_info.py

@@ -0,0 +1,41 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from ..._models import BaseModel
+
+__all__ = ["UsageInfo", "Cost"]
+
+
+class Cost(BaseModel):
+    input_tokens_cost: float
+
+    output_tokens_cost: float
+
+    total_cost: float
+
+    citation_tokens_cost: Optional[float] = None
+
+    reasoning_tokens_cost: Optional[float] = None
+
+    request_cost: Optional[float] = None
+
+    search_queries_cost: Optional[float] = None
+
+
+class UsageInfo(BaseModel):
+    completion_tokens: int
+
+    cost: Cost
+
+    prompt_tokens: int
+
+    total_tokens: int
+
+    citation_tokens: Optional[int] = None
+
+    num_search_queries: Optional[int] = None
+
+    reasoning_tokens: Optional[int] = None
+
+    search_context_size: Optional[str] = None

perplexity/types/shared_params/__init__.py

@@ -0,0 +1,4 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from .chat_message_input import ChatMessageInput as ChatMessageInput
+from .api_public_search_result import APIPublicSearchResult as APIPublicSearchResult

perplexity/types/shared_params/api_public_search_result.py

@@ -0,0 +1,22 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import Literal, Required, TypedDict
+
+__all__ = ["APIPublicSearchResult"]
+
+
+class APIPublicSearchResult(TypedDict, total=False):
+    title: Required[str]
+
+    url: Required[str]
+
+    date: Optional[str]
+
+    last_updated: Optional[str]
+
+    snippet: str
+
+    source: Literal["web", "attachment"]

perplexity/types/shared_params/chat_message_input.py

@@ -0,0 +1,178 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Union, Iterable, Optional
+from typing_extensions import Literal, Required, TypeAlias, TypedDict
+
+from ..._types import SequenceNotStr
+from .api_public_search_result import APIPublicSearchResult
+
+__all__ = [
+    "ChatMessageInput",
+    "ContentStructuredContent",
+    "ContentStructuredContentChatMessageContentTextChunk",
+    "ContentStructuredContentChatMessageContentImageChunk",
+    "ContentStructuredContentChatMessageContentImageChunkImageURL",
+    "ContentStructuredContentChatMessageContentImageChunkImageURLURL",
+    "ContentStructuredContentChatMessageContentFileChunk",
+    "ContentStructuredContentChatMessageContentFileChunkFileURL",
+    "ContentStructuredContentChatMessageContentFileChunkFileURLURL",
+    "ContentStructuredContentChatMessageContentPdfChunk",
+    "ContentStructuredContentChatMessageContentPdfChunkPdfURL",
+    "ContentStructuredContentChatMessageContentPdfChunkPdfURLURL",
+    "ContentStructuredContentChatMessageContentVideoChunk",
+    "ContentStructuredContentChatMessageContentVideoChunkVideoURL",
+    "ContentStructuredContentChatMessageContentVideoChunkVideoURLVideoURL",
+    "ReasoningStep",
+    "ReasoningStepExecutePython",
+    "ReasoningStepFetchURLContent",
+    "ReasoningStepWebSearch",
+    "ToolCall",
+    "ToolCallFunction",
+]
+
+
+class ContentStructuredContentChatMessageContentTextChunk(TypedDict, total=False):
+    text: Required[str]
+
+    type: Required[Literal["text"]]
+
+
+class ContentStructuredContentChatMessageContentImageChunkImageURLURL(TypedDict, total=False):
+    url: Required[str]
+
+
+ContentStructuredContentChatMessageContentImageChunkImageURL: TypeAlias = Union[
+    ContentStructuredContentChatMessageContentImageChunkImageURLURL, str
+]
+
+
+class ContentStructuredContentChatMessageContentImageChunk(TypedDict, total=False):
+    image_url: Required[ContentStructuredContentChatMessageContentImageChunkImageURL]
+
+    type: Required[Literal["image_url"]]
+
+
+class ContentStructuredContentChatMessageContentFileChunkFileURLURL(TypedDict, total=False):
+    url: Required[str]
+
+
+ContentStructuredContentChatMessageContentFileChunkFileURL: TypeAlias = Union[
+    ContentStructuredContentChatMessageContentFileChunkFileURLURL, str
+]
+
+
+class ContentStructuredContentChatMessageContentFileChunk(TypedDict, total=False):
+    file_url: Required[ContentStructuredContentChatMessageContentFileChunkFileURL]
+
+    type: Required[Literal["file_url"]]
+
+    file_name: Optional[str]
+
+
+class ContentStructuredContentChatMessageContentPdfChunkPdfURLURL(TypedDict, total=False):
+    url: Required[str]
+
+
+ContentStructuredContentChatMessageContentPdfChunkPdfURL: TypeAlias = Union[
+    ContentStructuredContentChatMessageContentPdfChunkPdfURLURL, str
+]
+
+
+class ContentStructuredContentChatMessageContentPdfChunk(TypedDict, total=False):
+    pdf_url: Required[ContentStructuredContentChatMessageContentPdfChunkPdfURL]
+
+    type: Required[Literal["pdf_url"]]
+
+
+class ContentStructuredContentChatMessageContentVideoChunkVideoURLVideoURL(TypedDict, total=False):
+    url: Required[str]
+
+    frame_interval: Union[str, int]
+
+
+ContentStructuredContentChatMessageContentVideoChunkVideoURL: TypeAlias = Union[
+    ContentStructuredContentChatMessageContentVideoChunkVideoURLVideoURL, str
+]
+
+
+class ContentStructuredContentChatMessageContentVideoChunk(TypedDict, total=False):
+    type: Required[Literal["video_url"]]
+
+    video_url: Required[ContentStructuredContentChatMessageContentVideoChunkVideoURL]
+
+
+ContentStructuredContent: TypeAlias = Union[
+    ContentStructuredContentChatMessageContentTextChunk,
+    ContentStructuredContentChatMessageContentImageChunk,
+    ContentStructuredContentChatMessageContentFileChunk,
+    ContentStructuredContentChatMessageContentPdfChunk,
+    ContentStructuredContentChatMessageContentVideoChunk,
+]
+
+
+class ReasoningStepExecutePython(TypedDict, total=False):
+    """Code generation step details wrapper class"""
+
+    code: Required[str]
+
+    result: Required[str]
+
+
+class ReasoningStepFetchURLContent(TypedDict, total=False):
+    """Fetch url content step details wrapper class"""
+
+    contents: Required[Iterable[APIPublicSearchResult]]
+
+
+class ReasoningStepWebSearch(TypedDict, total=False):
+    """Web search step details wrapper class"""
+
+    search_keywords: Required[SequenceNotStr[str]]
+
+    search_results: Required[Iterable[APIPublicSearchResult]]
+
+
+class ReasoningStep(TypedDict, total=False):
+    """Reasoning step wrapper class"""
+
+    thought: Required[str]
+
+    execute_python: Optional[ReasoningStepExecutePython]
+    """Code generation step details wrapper class"""
+
+    fetch_url_content: Optional[ReasoningStepFetchURLContent]
+    """Fetch url content step details wrapper class"""
+
+    type: Optional[str]
+
+    web_search: Optional[ReasoningStepWebSearch]
+    """Web search step details wrapper class"""
+
+
+class ToolCallFunction(TypedDict, total=False):
+    arguments: Optional[str]
+
+    name: Optional[str]
+
+
+class ToolCall(TypedDict, total=False):
+    id: Optional[str]
+
+    function: Optional[ToolCallFunction]
+
+    type: Optional[Literal["function"]]
+
+
+class ChatMessageInput(TypedDict, total=False):
+    content: Required[Union[str, Iterable[ContentStructuredContent], None]]
+
+    role: Required[Literal["system", "user", "assistant", "tool"]]
+    """Chat roles enum"""
+
+    reasoning_steps: Optional[Iterable[ReasoningStep]]
+
+    tool_call_id: Optional[str]
+
+    tool_calls: Optional[Iterable[ToolCall]]

perplexity/types/stream_chunk.py

@@ -0,0 +1,33 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+from typing_extensions import Literal
+
+from .._models import BaseModel
+from .shared.choice import Choice
+from .shared.usage_info import UsageInfo
+from .shared.api_public_search_result import APIPublicSearchResult
+
+__all__ = ["StreamChunk"]
+
+
+class StreamChunk(BaseModel):
+    id: str
+
+    choices: List[Choice]
+
+    created: int
+
+    model: str
+
+    citations: Optional[List[str]] = None
+
+    object: Optional[str] = None
+
+    search_results: Optional[List[APIPublicSearchResult]] = None
+
+    status: Optional[Literal["PENDING", "COMPLETED"]] = None
+
+    type: Optional[Literal["message", "info", "end_of_stream"]] = None
+
+    usage: Optional[UsageInfo] = None