latitude-sdk 0.1.0b3__py3-none-any.whl → 0.1.0b5__py3-none-any.whl

latitude_sdk/client/client.py

@@ -16,7 +16,7 @@ from latitude_sdk.sdk.errors import (
 from latitude_sdk.sdk.types import LogSources
 from latitude_sdk.util import Model
 
-RETRIABLE_STATUSES = [429, 500, 502, 503, 504]
+RETRIABLE_STATUSES = [408, 409, 429, 500, 502, 503, 504]
 
 ClientEvent = httpx_sse.ServerSentEvent
 

latitude_sdk/env/env.py

@@ -1,7 +1,7 @@
 from latitude_sdk.util import Model, get_env
 
 DEFAULT_GATEWAY_HOSTNAME = "gateway.latitude.so"
-DEFAULT_GATEWAY_PORT = 8787
+DEFAULT_GATEWAY_PORT = 443
 DEFAULT_GATEWAY_SSL = True
 
 

latitude_sdk/sdk/errors.py

@@ -57,3 +57,13 @@ class ApiError(Exception):
             return f"Unexpected API Error: {status} {message}"
 
         return message
+
+    def __eq__(self, other: object) -> bool:
+        return (
+            isinstance(other, ApiError)
+            and self.status == other.status
+            and self.code == other.code
+            and self.message == other.message
+            and self.response == other.response
+            and self.db_ref == other.db_ref
+        )

latitude_sdk/sdk/logs.py

@@ -1,4 +1,4 @@
-from typing import List, Optional
+from typing import Iterable, Optional
 
 from latitude_sdk.client import Client, CreateLogRequestBody, CreateLogRequestParams, RequestHandler
 from latitude_sdk.sdk.errors import ApiError, ApiErrorCodes
@@ -45,7 +45,7 @@ class Logs:
 
         return LogOptions(project_id=project_id, version_uuid=version_uuid)
 
-    async def create(self, path: str, messages: List[Message], options: CreateLogOptions) -> CreateLogResult:
+    async def create(self, path: str, messages: Iterable[Message], options: CreateLogOptions) -> CreateLogResult:
         log_options = self._ensure_options(options)
         options = CreateLogOptions(**{**dict(options), **dict(log_options)})
 
@@ -59,7 +59,7 @@ class Logs:
             ),
             body=CreateLogRequestBody(
                 path=path,
-                messages=messages,
+                messages=list(messages),
                 response=options.response,
             ),
         ) as response:

latitude_sdk/sdk/prompts.py

@@ -1,4 +1,4 @@
-from typing import Any, AsyncGenerator, Dict, List, Optional
+from typing import Any, AsyncGenerator, Dict, Iterable, List, Optional
 
 from latitude_sdk.client import (
     ChatPromptRequestBody,
@@ -119,7 +119,7 @@ class Prompts:
                 elif type == str(ChainEvents.Error):
                     event = ChainEventError.model_validate_json(stream_event.data)
                     raise ApiError(
-                        status=500,
+                        status=400,
                         code=ApiErrorCodes.AIRunError,
                         message=event.error.message,
                         response=stream_event.data,
@@ -130,7 +130,7 @@ class Prompts:
                         status=500,
                         code=ApiErrorCodes.InternalServerError,
                         message=f"Unknown latitude event: {type}",
-                        response=f"Unknown latitude event: {type}",
+                        response=stream_event.data,
                     )
 
             elif stream_event.event == str(StreamEvents.Provider):
@@ -142,7 +142,7 @@ class Prompts:
                     status=500,
                     code=ApiErrorCodes.InternalServerError,
                     message=f"Unknown stream event: {stream_event.event}",
-                    response=f"Unknown stream event: {stream_event.event}",
+                    response=stream_event.data,
                 )
 
             if callbacks.on_event:
@@ -247,7 +247,9 @@ class Prompts:
 
             return None
 
-    async def chat(self, uuid: str, messages: List[Message], options: ChatPromptOptions) -> Optional[ChatPromptResult]:
+    async def chat(
+        self, uuid: str, messages: Iterable[Message], options: ChatPromptOptions
+    ) -> Optional[ChatPromptResult]:
         try:
             async with self._client.request(
                 handler=RequestHandler.ChatPrompt,
@@ -255,7 +257,7 @@ class Prompts:
                     conversation_uuid=uuid,
                 ),
                 body=ChatPromptRequestBody(
-                    messages=messages,
+                    messages=list(messages),
                     stream=options.stream,
                 ),
             ) as response:

latitude_sdk/sdk/types.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import Any, Dict, List, Optional, Protocol, Union, runtime_checkable
+from typing import Any, Dict, List, Literal, Optional, Protocol, Union, runtime_checkable
 
 from latitude_sdk.sdk.errors import ApiError
 from latitude_sdk.util import Field, Model, StrEnum
@@ -37,32 +37,32 @@ class ContentType(StrEnum):
 
 
 class TextContent(Model):
-    type: ContentType = ContentType.Text
+    type: Literal[ContentType.Text] = ContentType.Text
     text: str
 
 
 class ImageContent(Model):
-    type: ContentType = ContentType.Image
+    type: Literal[ContentType.Image] = ContentType.Image
     image: str
 
 
 class FileContent(Model):
-    type: ContentType = ContentType.File
+    type: Literal[ContentType.File] = ContentType.File
     file: str
     mime_type: str = Field(alias=str("mimeType"))
 
 
 class ToolCallContent(Model):
-    type: ContentType = ContentType.ToolCall
-    tool_call_id: str = Field(alias=str("toolCallId"))
-    tool_name: str = Field(alias=str("toolName"))
-    args: Dict[str, Any]
+    type: Literal[ContentType.ToolCall] = ContentType.ToolCall
+    id: str = Field(alias=str("toolCallId"))
+    name: str = Field(alias=str("toolName"))
+    arguments: Dict[str, Any] = Field(alias=str("args"))
 
 
 class ToolResultContent(Model):
-    type: ContentType = ContentType.ToolResult
-    tool_call_id: str = Field(alias=str("toolCallId"))
-    tool_name: str = Field(alias=str("toolName"))
+    type: Literal[ContentType.ToolResult] = ContentType.ToolResult
+    id: str = Field(alias=str("toolCallId"))
+    name: str = Field(alias=str("toolName"))
     result: str
     is_error: Optional[bool] = Field(default=None, alias=str("isError"))
 
@@ -85,30 +85,23 @@ class MessageRole(StrEnum):
 
 
 class SystemMessage(Model):
-    role: MessageRole = MessageRole.System
+    role: Literal[MessageRole.System] = MessageRole.System
     content: Union[str, List[TextContent]]
 
 
 class UserMessage(Model):
-    role: MessageRole = MessageRole.User
+    role: Literal[MessageRole.User] = MessageRole.User
     content: Union[str, List[Union[TextContent, ImageContent, FileContent]]]
     name: Optional[str] = None
 
 
-class ToolCall(Model):
-    id: str
-    name: str
-    arguments: Dict[str, Any]
-
-
 class AssistantMessage(Model):
-    role: MessageRole = MessageRole.Assistant
+    role: Literal[MessageRole.Assistant] = MessageRole.Assistant
     content: Union[str, List[Union[TextContent, ToolCallContent]]]
-    tool_calls: Optional[List[ToolCall]] = Field(default=None, alias=str("toolCalls"))
 
 
 class ToolMessage(Model):
-    role: MessageRole = MessageRole.Tool
+    role: Literal[MessageRole.Tool] = MessageRole.Tool
     content: List[ToolResultContent]
 
 
@@ -121,20 +114,26 @@ class ModelUsage(Model):
     total_tokens: int = Field(alias=str("totalTokens"))
 
 
+class ToolCall(Model):
+    id: str
+    name: str
+    arguments: Dict[str, Any]
+
+
 class StreamTypes(StrEnum):
     Text = "text"
     Object = "object"
 
 
 class ChainTextResponse(Model):
-    type: StreamTypes = StreamTypes.Text
+    type: Literal[StreamTypes.Text] = Field(default=StreamTypes.Text, alias=str("streamType"))
     text: str
     tool_calls: Optional[List[ToolCall]] = Field(default=None, alias=str("toolCalls"))
     usage: ModelUsage
 
 
 class ChainObjectResponse(Model):
-    type: StreamTypes = StreamTypes.Object
+    type: Literal[StreamTypes.Object] = Field(default=StreamTypes.Object, alias=str("streamType"))
     object: Any
     usage: ModelUsage
 
@@ -165,34 +164,34 @@ class ChainEvents(StrEnum):
 
 
 class ChainEventStep(Model):
-    event: StreamEvents = StreamEvents.Latitude
-    type: ChainEvents = ChainEvents.Step
-    config: Dict[str, Any]
+    event: Literal[StreamEvents.Latitude] = StreamEvents.Latitude
+    type: Literal[ChainEvents.Step] = ChainEvents.Step
+    uuid: Optional[str] = None
     is_last_step: bool = Field(alias=str("isLastStep"))
+    config: Dict[str, Any]
     messages: List[Message]
-    uuid: Optional[str] = None
 
 
 class ChainEventStepCompleted(Model):
-    event: StreamEvents = StreamEvents.Latitude
-    type: ChainEvents = ChainEvents.StepCompleted
-    response: ChainResponse
+    event: Literal[StreamEvents.Latitude] = StreamEvents.Latitude
+    type: Literal[ChainEvents.StepCompleted] = ChainEvents.StepCompleted
     uuid: Optional[str] = None
+    response: ChainResponse
 
 
 class ChainEventCompleted(Model):
-    event: StreamEvents = StreamEvents.Latitude
-    type: ChainEvents = ChainEvents.Completed
+    event: Literal[StreamEvents.Latitude] = StreamEvents.Latitude
+    type: Literal[ChainEvents.Completed] = ChainEvents.Completed
+    uuid: Optional[str] = None
     config: Dict[str, Any]
     messages: Optional[List[Message]] = None
     object: Optional[Any] = None
     response: ChainResponse
-    uuid: Optional[str] = None
 
 
 class ChainEventError(Model):
-    event: StreamEvents = StreamEvents.Latitude
-    type: ChainEvents = ChainEvents.Error
+    event: Literal[StreamEvents.Latitude] = StreamEvents.Latitude
+    type: Literal[ChainEvents.Error] = ChainEvents.Error
     error: ChainError
 
 
@@ -203,7 +202,7 @@ LatitudeEvent = ChainEvent
 
 
 class FinishedEvent(Model):
-    event: StreamEvents = StreamEvents.Finished
+    event: Literal[StreamEvents.Finished] = StreamEvents.Finished
     uuid: str
     conversation: List[Message]
     response: ChainResponse

{latitude_sdk-0.1.0b3.dist-info → latitude_sdk-0.1.0b5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: latitude-sdk
-Version: 0.1.0b3
+Version: 0.1.0b5
 Summary: Latitude SDK for Python
 Project-URL: repository, https://github.com/latitude-dev/latitude-llm/tree/main/packages/sdks/python
 Project-URL: homepage, https://github.com/latitude-dev/latitude-llm/tree/main/packages/sdks/python#readme

{latitude_sdk-0.1.0b3.dist-info → latitude_sdk-0.1.0b5.dist-info}/RECORD

@@ -1,20 +1,20 @@
 latitude_sdk/__init__.py,sha256=-AbNXLmzDZeGbRdDIOpNjdCbacOvLBflSJwQtLlZfgk,19
 latitude_sdk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 latitude_sdk/client/__init__.py,sha256=d8CnNB8UoGwcftiIeeC0twdg01qNvfpj-v7O40I7IiE,68
-latitude_sdk/client/client.py,sha256=ff49AlX6hxfoAwKA4_c5B6zXuZMQwV588_XaD6A7K4E,4379
+latitude_sdk/client/client.py,sha256=Oc4COkVFR1vFewVKZzUIvztJi_yTxeSMoyeML-ivVsY,4389
 latitude_sdk/client/payloads.py,sha256=4fZS5jlTRRiA93QiDMUBh1utctI_9Xyh0fQV2bygiRU,2696
 latitude_sdk/client/router.py,sha256=XYi24oUKL1wHDLvydpwLWCjy9tDGSVPN-zhetuvaj8I,4038
 latitude_sdk/env/__init__.py,sha256=66of5veJ-u1aNI025L65Rrj321AjrYevMqomTMYIrPQ,19
-latitude_sdk/env/env.py,sha256=OSBVGHXJr_48plkVagCANNy5n5F3tb48Dug_r7uo0m4,456
+latitude_sdk/env/env.py,sha256=MnXexPOHE6aXcAszrDCbW7hzACUv4YtU1bfxpYwvHNw,455
 latitude_sdk/sdk/__init__.py,sha256=C9LlIjfnrS7KOK3-ruXKmbT77nSQMm23nZ6-t8sO8ME,137
-latitude_sdk/sdk/errors.py,sha256=cq5swDrRFYAZWtohR1Fz9aq0D17eSb2JUTa-FeTNOC8,1412
+latitude_sdk/sdk/errors.py,sha256=9GlGdDE8LGy3dE2Ry_BipBg-tDbQx7LWXJfSnTJSSBE,1747
 latitude_sdk/sdk/evaluations.py,sha256=ASWfNfH124qeahzhAn-gb2Ep4QIew5uDveY5NbNsNfk,2086
 latitude_sdk/sdk/latitude.py,sha256=8OZ_8st4EkaUvyvmzFBYGtnkFQzFtaRE_OyY3iQes_4,2471
-latitude_sdk/sdk/logs.py,sha256=BXo-SN-CarRmR_HHk7biutqyegA01UkjkIKN0Huz4Zc,2042
-latitude_sdk/sdk/prompts.py,sha256=HiFlOl56YOwR_Kq9ce6o4fNsoQwJucFYY9DrtUqCjMo,10144
-latitude_sdk/sdk/types.py,sha256=-y6aH8TK-dAOqC-ozzBp7Kzk7vxANkyFxqVWmDLQE5A,7527
+latitude_sdk/sdk/logs.py,sha256=UBlQzNq8AtWVkeVztqUWTs78jGkfA8a2r6wOhasUpxs,2056
+latitude_sdk/sdk/prompts.py,sha256=VaqI3qbVpcSbiT5xtd63eEwbIGTdMusDSoZeQZO8z4g,10134
+latitude_sdk/sdk/types.py,sha256=1sBtKNppipuayPj3j8qoUzMekIO0DxSBBXYNCEVsZjU,7866
 latitude_sdk/util/__init__.py,sha256=alIDGBnxWH4JvP-UW-7N99seBBi0r1GV1h8f1ERFBec,21
 latitude_sdk/util/utils.py,sha256=WKm20UsceP352psbP8bmlgZ_IPTPYKXUIrXcesG_c08,2710
-latitude_sdk-0.1.0b3.dist-info/METADATA,sha256=nLZCboyGmJwZSc4WwT43kJddtoeykt-KQG0WmkpG2bk,1982
-latitude_sdk-0.1.0b3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-latitude_sdk-0.1.0b3.dist-info/RECORD,,
+latitude_sdk-0.1.0b5.dist-info/METADATA,sha256=WiPG0_TK3U1Ojeo1MKo47hqz3imEzLn1pG2PwJH7otk,1982
+latitude_sdk-0.1.0b5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+latitude_sdk-0.1.0b5.dist-info/RECORD,,