chore: Move ti queries directory

julep-ai · Dec 19, 2024 · a7d10da · a7d10da
1 parent 7dd2bf0
commit a7d10da
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 44 deletions.
diff --git a/agents-api/agents_api/models/chat/get_cached_response.py b/agents-api/agents_api/models/chat/get_cached_response.py
diff --git a/agents-api/agents_api/models/chat/set_cached_response.py b/agents-api/agents_api/models/chat/set_cached_response.py
diff --git a/...ts-api/agents_api/models/chat/__init__.py → ...s-api/agents_api/queries/chat/__init__.py b/...ts-api/agents_api/models/chat/__init__.py → ...s-api/agents_api/queries/chat/__init__.py
@@ -17,6 +17,4 @@
 # ruff: noqa: F401, F403, F405
 
 from .gather_messages import gather_messages
-from .get_cached_response import get_cached_response
 from .prepare_chat_context import prepare_chat_context
-from .set_cached_response import set_cached_response
diff --git a/...agents_api/models/chat/gather_messages.py → ...gents_api/queries/chat/gather_messages.py b/...agents_api/models/chat/gather_messages.py → ...gents_api/queries/chat/gather_messages.py
diff --git a/...s_api/models/chat/prepare_chat_context.py → ..._api/queries/chat/prepare_chat_context.py b/...s_api/models/chat/prepare_chat_context.py → ..._api/queries/chat/prepare_chat_context.py
@@ -3,7 +3,6 @@
 
 from beartype import beartype
 from fastapi import HTTPException
-from pycozo.client import QueryException
 from pydantic import ValidationError
 
 from ...common.protocol.sessions import ChatContext, make_session
@@ -22,13 +21,13 @@
 T = TypeVar("T")
 
 
-@rewrap_exceptions(
-    {
-        QueryException: partialclass(HTTPException, status_code=400),
-        ValidationError: partialclass(HTTPException, status_code=400),
-        TypeError: partialclass(HTTPException, status_code=400),
-    }
-)
+# TODO: implement this part
+# @rewrap_exceptions(
+#     {
+#         ValidationError: partialclass(HTTPException, status_code=400),
+#         TypeError: partialclass(HTTPException, status_code=400),
+#     }
+# )
 @wrap_in_class(
     ChatContext,
     one=True,