edison-client 0.10.1.dev351__tar.gz → 0.10.1.dev352__tar.gz

{edison_client-0.10.1.dev351/src/edison_client.egg-info → edison_client-0.10.1.dev352}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: edison-client
-Version: 0.10.1.dev351
+Version: 0.10.1.dev352
 Summary: A client for interacting with endpoints of the Edison Scientific service.
 Author-email: Edison Scientific technical staff <hello@futurehouse.org>
 License:                                  Apache License

{edison_client-0.10.1.dev351 → edison_client-0.10.1.dev352}/src/edison_client/clients/chat_methods.py

@@ -19,7 +19,11 @@ from edison_client.models.chat import (
     ConversationDetail,
     StoreMessagePayload,
 )
-from edison_client.utils.general import retry_if_connection_error
+from edison_client.utils.general import (
+    _BASE_CONNECTION_ERRORS,
+    RETRYABLE_HTTP_STATUS_CODES,
+    retry_if_connection_error,
+)
 
 logger = logging.getLogger(__name__)
 
@@ -49,13 +53,22 @@ class ChatMethods:
             "async_client property must be implemented by subclass"
         )
 
-    def _handle_chat_http_errors(self, e: HTTPStatusError, operation: str) -> NoReturn:  # noqa: PLR6301
+    def _handle_chat_http_errors(  # noqa: PLR6301
+        self,
+        e: HTTPStatusError,
+        operation: str,
+        exc_class: type[ChatError] = ChatError,
+    ) -> NoReturn:
         """Handle common HTTP errors for chat operations."""
+        if e.response.status_code in RETRYABLE_HTTP_STATUS_CODES:
+            raise e
         if e.response.status_code == codes.FORBIDDEN:
-            raise ChatError(f"Error {operation}: not authorized") from e
+            raise exc_class(
+                f"Error {operation}: {e.response.status_code} - {e.response.text}"
+            ) from e
         if e.response.status_code == codes.UNPROCESSABLE_ENTITY:
-            raise ChatError(f"Invalid request payload: {e.response.text}") from e
-        raise ChatError(
+            raise exc_class(f"Invalid request payload: {e.response.text}") from e
+        raise exc_class(
             f"Error {operation}: {e.response.status_code} - {e.response.text}"
         ) from e
 
@@ -108,6 +121,8 @@ class ChatMethods:
         except ChatError:
             raise
         except Exception as e:
+            if isinstance(e, _BASE_CONNECTION_ERRORS):
+                raise
             raise ChatError(f"An unexpected error occurred: {e!r}") from e
 
     @retry(
@@ -159,6 +174,8 @@ class ChatMethods:
         except ChatError:
             raise
         except Exception as e:
+            if isinstance(e, _BASE_CONNECTION_ERRORS):
+                raise
             raise ChatError(f"An unexpected error occurred: {e!r}") from e
 
     @retry(
@@ -203,6 +220,8 @@ class ChatMethods:
         except ChatError:
             raise
         except Exception as e:
+            if isinstance(e, _BASE_CONNECTION_ERRORS):
+                raise
             raise ChatError(f"An unexpected error occurred: {e!r}") from e
 
     @retry(
@@ -247,6 +266,8 @@ class ChatMethods:
         except ChatError:
             raise
         except Exception as e:
+            if isinstance(e, _BASE_CONNECTION_ERRORS):
+                raise
             raise ChatError(f"An unexpected error occurred: {e!r}") from e
 
     @retry(
@@ -283,10 +304,12 @@ class ChatMethods:
             response.raise_for_status()
             return ChatConversationList.model_validate(response.json())
         except HTTPStatusError as e:
-            self._handle_chat_http_errors(e, "fetching conversations")
+            self._handle_chat_http_errors(e, "fetching conversations", ChatFetchError)
         except ChatError:
             raise
         except Exception as e:
+            if isinstance(e, _BASE_CONNECTION_ERRORS):
+                raise
             raise ChatFetchError(f"An unexpected error occurred: {e!r}") from e
 
     @retry(
@@ -323,10 +346,12 @@ class ChatMethods:
             response.raise_for_status()
             return ChatConversationList.model_validate(response.json())
         except HTTPStatusError as e:
-            self._handle_chat_http_errors(e, "fetching conversations")
+            self._handle_chat_http_errors(e, "fetching conversations", ChatFetchError)
         except ChatError:
             raise
         except Exception as e:
+            if isinstance(e, _BASE_CONNECTION_ERRORS):
+                raise
             raise ChatFetchError(f"An unexpected error occurred: {e!r}") from e
 
     @retry(
@@ -352,10 +377,12 @@ class ChatMethods:
             response.raise_for_status()
             return ConversationDetail.model_validate(response.json())
         except HTTPStatusError as e:
-            self._handle_chat_http_errors(e, "fetching conversation")
+            self._handle_chat_http_errors(e, "fetching conversation", ChatFetchError)
         except ChatError:
             raise
         except Exception as e:
+            if isinstance(e, _BASE_CONNECTION_ERRORS):
+                raise
             raise ChatFetchError(f"An unexpected error occurred: {e!r}") from e
 
     @retry(
@@ -381,10 +408,12 @@ class ChatMethods:
             response.raise_for_status()
             return ConversationDetail.model_validate(response.json())
         except HTTPStatusError as e:
-            self._handle_chat_http_errors(e, "fetching conversation")
+            self._handle_chat_http_errors(e, "fetching conversation", ChatFetchError)
         except ChatError:
             raise
         except Exception as e:
+            if isinstance(e, _BASE_CONNECTION_ERRORS):
+                raise
             raise ChatFetchError(f"An unexpected error occurred: {e!r}") from e
 
     # ── store_conversation_message ─────────────────────────────────────
@@ -427,6 +456,8 @@ class ChatMethods:
         except ChatError:
             raise
         except Exception as e:
+            if isinstance(e, _BASE_CONNECTION_ERRORS):
+                raise
             raise ChatError(f"An unexpected error occurred: {e!r}") from e
 
     @retry(
@@ -467,4 +498,6 @@ class ChatMethods:
         except ChatError:
             raise
         except Exception as e:
+            if isinstance(e, _BASE_CONNECTION_ERRORS):
+                raise
             raise ChatError(f"An unexpected error occurred: {e!r}") from e

{edison_client-0.10.1.dev351 → edison_client-0.10.1.dev352}/src/edison_client/models/chat.py

@@ -1,3 +1,4 @@
+from datetime import datetime
 from enum import StrEnum, auto
 from typing import Any
 from uuid import UUID
@@ -89,8 +90,8 @@ class ChatConversation(BaseModel):
     """A conversation grouped by session."""
 
     session_id: UUID = Field(description="The session ID for this conversation")
-    created_at: str = Field(
-        description="ISO timestamp when the conversation was created"
+    created_at: datetime = Field(
+        description="Timestamp when the conversation was created"
     )
     messages: list[Any] = Field(description="The messages in this conversation")
 
@@ -118,9 +119,6 @@ class StoreMessagePayload(BaseModel):
     """Payload for storing a conversation message."""
 
     message: dict = Field(description="Message JSON (aviary Message format)")
-    session_id: str | None = Field(
-        default=None, description="Session ID for the conversation"
-    )
     metadata: dict | None = Field(
         default=None, description="Optional metadata to attach to the message"
     )

{edison_client-0.10.1.dev351 → edison_client-0.10.1.dev352}/src/edison_client/version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.10.1.dev351'
-__version_tuple__ = version_tuple = (0, 10, 1, 'dev351')
+__version__ = version = '0.10.1.dev352'
+__version_tuple__ = version_tuple = (0, 10, 1, 'dev352')
 
-__commit_id__ = commit_id = 'g1a73d2e50'
+__commit_id__ = commit_id = 'gbbd65e731'

{edison_client-0.10.1.dev351 → edison_client-0.10.1.dev352/src/edison_client.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: edison-client
-Version: 0.10.1.dev351
+Version: 0.10.1.dev352
 Summary: A client for interacting with endpoints of the Edison Scientific service.
 Author-email: Edison Scientific technical staff <hello@futurehouse.org>
 License:                                  Apache License

{edison_client-0.10.1.dev351 → edison_client-0.10.1.dev352}/tests/test_chat_methods.py

@@ -4,7 +4,7 @@ from uuid import UUID, uuid4
 
 import pytest
 
-from edison_client.clients.chat_methods import ChatError
+from edison_client.clients.chat_methods import ChatError, ChatFetchError
 from edison_client.clients.rest_client import RestClient
 from edison_client.models.chat import (
     ChatConversationList,
@@ -120,7 +120,7 @@ class TestSyncChatLifecycle:
 
     def test_get_conversation_not_found(self, admin_client: RestClient):
         """Fetching a non-existent session returns 404."""
-        with pytest.raises(ChatError, match="404") as exc_info:
+        with pytest.raises(ChatFetchError, match="404") as exc_info:
             admin_client.get_conversation(uuid4())
         assert "not found" in str(exc_info.value).lower()
 
@@ -233,7 +233,7 @@ class TestAsyncChatLifecycle:
     @pytest.mark.asyncio
     async def test_aget_conversation_not_found(self, admin_client: RestClient):
         """Fetching a non-existent session returns 404 (async)."""
-        with pytest.raises(ChatError, match="404") as exc_info:
+        with pytest.raises(ChatFetchError, match="404") as exc_info:
             await admin_client.aget_conversation(uuid4())
         assert "not found" in str(exc_info.value).lower()