snowflake-cortex-agent-sdk 0.0.1__py3-none-any.whl

cortex_agent_sdk/__init__.py

@@ -0,0 +1,180 @@
+"""Snowflake Cortex Agent SDK (Python).
+
+Sync::
+
+    from cortex_agent_sdk import Conversation, CortexAgentClient
+
+    client = CortexAgentClient(account="myacct", auth="<pat>")
+    convo = Conversation.create(client)
+    print(convo.ask("What is Snowflake Arctic?").text)
+
+Async::
+
+    import asyncio
+    from cortex_agent_sdk import AsyncConversation, AsyncCortexAgentClient
+
+    async def main():
+        async with AsyncCortexAgentClient(account="myacct", auth="<pat>") as client:
+            convo = await AsyncConversation.create(client)
+            print((await convo.ask("What is Snowflake Arctic?")).text)
+
+    asyncio.run(main())
+
+Start with ``docs/01-getting-started.md``.
+"""
+
+from __future__ import annotations
+
+from . import models
+from ._constants import DEFAULT_MAX_RETRIES, DEFAULT_TIMEOUT
+from ._request_options import RequestOptions
+from ._transport import AuthRefreshContext, RetryPolicy
+from ._types import NOT_GIVEN, NotGiven, Omit, not_given, omit
+from ._version import __title__, __version__
+from .api_response import APIResponse, ResponseCarrier
+from .auth import (
+    Authenticator,
+    CallableAuth,
+    KeyPairJwtAuth,
+    OAuthAuth,
+    PatAuth,
+    SnowflakeSessionTokenAuth,
+    SnowflakeTokenAuth,
+    SnowflakeTokenType,
+)
+from .client import AsyncCortexAgentClient, CortexAgentClient
+from .constants import ContentType, ContentTypeName, EventType, EventTypeName
+from .conversation import AsyncConversation, Conversation
+from .errors import (
+    APIConnectionAbortedError,
+    APIConnectionError,
+    APIStatusError,
+    APITimeoutError,
+    AuthenticationError,
+    BadRequestError,
+    ConflictError,
+    CortexAgentAPIError,
+    CortexAgentError,
+    InternalServerError,
+    NotFoundError,
+    PermissionDeniedError,
+    RateLimitError,
+    UnprocessableEntityError,
+    create_api_error,
+)
+from .resources.agent import (
+    AgentResource,
+    AgentSchemaTarget,
+    AsyncAgentResource,
+    CreateMode,
+    DataAgentTarget,
+    DataAgentVersionTarget,
+)
+from .resources.coding_agent import (
+    AsyncCodingAgentResource,
+    CodingAgentPermissionPolicy,
+    CodingAgentResource,
+    expand_coding_agent_request,
+)
+from .resources.feedback import AsyncFeedbackResource, FeedbackResource
+from .resources.runs import AsyncRunsResource, RunsResource
+from .resources.threads import AsyncThreadsResource, ThreadsResource
+from .streaming.parser import AgentEvent, aiter_sse, iter_sse
+from .streaming.run_stream import AgentRunStream, AsyncAgentRunStream, StreamEvent
+from .threads_page import AsyncThreadsPage, ThreadsPage
+from .views.agent_response import AgentResponseView
+from .views.async_run import AsyncBackgroundRun, AsyncRun, BackgroundRun
+from .views.content_view import MessageContentView
+from .views.thread_message import ThreadMessage, ThreadMessageView
+
+__all__ = [
+    # Core clients + config
+    "APIResponse",
+    "AuthRefreshContext",
+    "AsyncCortexAgentClient",
+    "CortexAgentClient",
+    "DEFAULT_MAX_RETRIES",
+    "DEFAULT_TIMEOUT",
+    "RequestOptions",
+    "ResponseCarrier",
+    "RetryPolicy",
+    # Sentinels
+    "NOT_GIVEN",
+    "NotGiven",
+    "Omit",
+    "not_given",
+    "omit",
+    # Auth
+    "Authenticator",
+    "CallableAuth",
+    "KeyPairJwtAuth",
+    "OAuthAuth",
+    "PatAuth",
+    "SnowflakeSessionTokenAuth",
+    "SnowflakeTokenAuth",
+    "SnowflakeTokenType",
+    # Errors
+    "APIConnectionAbortedError",
+    "APIConnectionError",
+    "APIStatusError",
+    "APITimeoutError",
+    "AuthenticationError",
+    "BadRequestError",
+    "ConflictError",
+    "CortexAgentAPIError",
+    "CortexAgentError",
+    "InternalServerError",
+    "NotFoundError",
+    "PermissionDeniedError",
+    "RateLimitError",
+    "UnprocessableEntityError",
+    "create_api_error",
+    # Discriminator constants
+    "ContentType",
+    "ContentTypeName",
+    "EventType",
+    "EventTypeName",
+    # Streaming
+    "AgentEvent",
+    "AgentRunStream",
+    "AsyncAgentRunStream",
+    "StreamEvent",
+    "aiter_sse",
+    "iter_sse",
+    # Pagination
+    "AsyncThreadsPage",
+    "ThreadsPage",
+    # Resources
+    "AgentResource",
+    "AgentSchemaTarget",
+    "AsyncAgentResource",
+    "AsyncCodingAgentResource",
+    "AsyncFeedbackResource",
+    "AsyncRunsResource",
+    "AsyncThreadsResource",
+    "CodingAgentPermissionPolicy",
+    "CodingAgentResource",
+    "CreateMode",
+    "DataAgentTarget",
+    "DataAgentVersionTarget",
+    "FeedbackResource",
+    "RunsResource",
+    "ThreadsResource",
+    "expand_coding_agent_request",
+    # Response / message views
+    "AgentResponseView",
+    "AsyncBackgroundRun",
+    "AsyncRun",
+    "BackgroundRun",
+    "MessageContentView",
+    "ThreadMessage",
+    "ThreadMessageView",
+    # Conversation
+    "AsyncConversation",
+    "Conversation",
+    # Generated models
+    "models",
+    # Metadata
+    "__title__",
+    "__version__",
+]