fix: Apply fixes

agents-api/agents_api/autogen/Chat.py

@@ -6,7 +6,7 @@
 from typing import Annotated, Literal
 from uuid import UUID
 
-from pydantic import AwareDatetime, BaseModel, ConfigDict, Field
+from pydantic import AwareDatetime, BaseModel, ConfigDict, Field, RootModel
 
 from .Docs import DocReference
 from .Entries import ChatMLMessage

agents-api/agents_api/autogen/Entries.py

@@ -6,7 +6,7 @@
 from typing import Annotated, Literal
 from uuid import UUID
 
-from pydantic import AnyUrl, AwareDatetime, BaseModel, ConfigDict, Field
+from pydantic import AnyUrl, AwareDatetime, BaseModel, ConfigDict, Field, RootModel
 
 from .Tools import ChosenToolCall, Tool, ToolResponse
 

agents-api/agents_api/models/agent/create_agent.py

@@ -65,7 +65,7 @@ def create_agent(
         if isinstance(data.instructions, list)
         else [data.instructions]
     )
-    data.default_settings = data.default_settings or {}
+    data.default_settings = data.default_settings
 
     agent_data = data.model_dump(exclude_unset=True)
     default_settings = agent_data.pop("default_settings")

agents-api/agents_api/models/agent/update_agent.py

@@ -49,7 +49,11 @@ def update_agent(
     Returns:
     ResourceUpdatedResponse: The updated agent data.
     """
-    default_settings = data.default_settings.model_dump(exclude_none=True)
+    default_settings = (
+        data.default_settings.model_dump(exclude_none=True)
+        if data.default_settings
+        else {}
+    )
     update_data = data.model_dump()
 
     # Remove default settings from the agent update data

agents-api/agents_api/routers/agents/create_agent_tool.py

@@ -25,6 +25,6 @@ async def create_agent_tool(
         developer_id=x_developer_id,
         agent_id=agent_id,
         data=[data],
-    )
+    )[0]
 
     return ResourceCreatedResponse(id=tool.id, created_at=tool.created_at)

agents-api/agents_api/routers/sessions/create_session.py

@@ -26,4 +26,5 @@ async def create_session(
     return ResourceCreatedResponse(
         id=session.id,
         created_at=session.created_at,
+        jobs=[],
     )

agents-api/agents_api/routers/sessions/get_session_history.py

@@ -3,7 +3,7 @@
 from fastapi import Depends
 from pydantic import UUID4
 
-from ...autogen.openapi_model import Session
+from ...autogen.openapi_model import History
 from ...dependencies.developer_id import get_developer_id
 from ...models.entry.get_history import get_history as get_history_query
 from .router import router
@@ -12,5 +12,5 @@
 @router.get("/sessions/{session_id}/history", tags=["sessions"])
 async def get_session_history(
     session_id: UUID4, x_developer_id: Annotated[UUID4, Depends(get_developer_id)]
-) -> Session:
+) -> History:
     return get_history_query(developer_id=x_developer_id, session_id=session_id)

agents-api/agents_api/routers/users/create_or_update_user.py

@@ -24,4 +24,4 @@ async def create_or_update_user(
         data=data,
     )
 
-    return ResourceCreatedResponse(id=user.id, created_at=user.created_at)
+    return ResourceCreatedResponse(id=user.id, created_at=user.created_at, jobs=[])

agents-api/agents_api/routers/users/update_user.py

@@ -18,5 +18,5 @@ async def update_user(
     return update_user_query(
         developer_id=x_developer_id,
         user_id=user_id,
-        data=data,
+        update_user=data,
     )

agents-api/agents_api/web.py

@@ -8,7 +8,7 @@
 import sentry_sdk
 import uvicorn
 from fastapi import Depends, FastAPI, Request, status
-from fastapi.exceptions import RequestValidationError
+from fastapi.exceptions import HTTPException, RequestValidationError
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
 from litellm.exceptions import APIError
@@ -21,6 +21,8 @@
 from agents_api.exceptions import PromptTooBigError
 from agents_api.routers import (
     agents,
+    jobs,
+    sessions,
     tasks,
     users,
 )
@@ -88,12 +90,20 @@ def register_exceptions(app: FastAPI):
 register_exceptions(app)
 
 app.include_router(agents.router)
-# app.include_router(sessions.router)
+app.include_router(sessions.router)
 app.include_router(users.router)
-# app.include_router(jobs.router)
+app.include_router(jobs.router)
 app.include_router(tasks.router)
 
 
+@app.exception_handler(HTTPException)
+async def http_exception_handler(request, exc: HTTPException):  # pylint: disable=unused-argument
+    return JSONResponse(
+        status_code=exc.status_code,
+        content={"error": {"message": str(exc)}},
+    )
+
+
 @app.exception_handler(RPCError)
 async def validation_error_handler(request: Request, exc: RPCError):
     return JSONResponse(

scripts/generate_openapi_code.sh

@@ -7,7 +7,7 @@ cd typespec/ && \
     tsp compile .
 cd -
 
-fern generate --local
+fern generate
 
 cd sdks/python && \
     poetry update && \

sdks/python/julep/api/__init__.py

@@ -195,7 +195,6 @@
     ToolsChosenToolCall_Function,
     ToolsFunctionCallOption,
     ToolsFunctionDef,
-    ToolsFunctionDefUpdate,
     ToolsFunctionTool,
     ToolsNamedFunctionChoice,
     ToolsNamedToolChoice,
@@ -412,7 +411,6 @@
     "ToolsChosenToolCall_Function",
     "ToolsFunctionCallOption",
     "ToolsFunctionDef",
-    "ToolsFunctionDefUpdate",
     "ToolsFunctionTool",
     "ToolsNamedFunctionChoice",
     "ToolsNamedToolChoice",

sdks/python/julep/api/client.py

@@ -98,7 +98,6 @@
 from .types.tasks_task_tool import TasksTaskTool
 from .types.tasks_update_task_request_main_item import TasksUpdateTaskRequestMainItem
 from .types.tools_function_def import ToolsFunctionDef
-from .types.tools_function_def_update import ToolsFunctionDefUpdate
 from .types.tools_tool_type import ToolsToolType
 from .types.user_docs_route_list_request_direction import (
     UserDocsRouteListRequestDirection,
@@ -1585,7 +1584,7 @@ def agent_tools_route_patch(
         *,
         type: typing.Optional[ToolsToolType] = OMIT,
         name: typing.Optional[CommonValidPythonIdentifier] = OMIT,
-        function: typing.Optional[ToolsFunctionDefUpdate] = OMIT,
+        function: typing.Optional[ToolsFunctionDef] = OMIT,
         integration: typing.Optional[typing.Any] = OMIT,
         system: typing.Optional[typing.Any] = OMIT,
         api_call: typing.Optional[typing.Any] = OMIT,
@@ -1608,7 +1607,7 @@ def agent_tools_route_patch(
         name : typing.Optional[CommonValidPythonIdentifier]
             Name of the tool (must be unique for this agent and a valid python identifier string )
 
-        function : typing.Optional[ToolsFunctionDefUpdate]
+        function : typing.Optional[ToolsFunctionDef]
 
         integration : typing.Optional[typing.Any]
 
@@ -5177,7 +5176,7 @@ async def agent_tools_route_patch(
         *,
         type: typing.Optional[ToolsToolType] = OMIT,
         name: typing.Optional[CommonValidPythonIdentifier] = OMIT,
-        function: typing.Optional[ToolsFunctionDefUpdate] = OMIT,
+        function: typing.Optional[ToolsFunctionDef] = OMIT,
         integration: typing.Optional[typing.Any] = OMIT,
         system: typing.Optional[typing.Any] = OMIT,
         api_call: typing.Optional[typing.Any] = OMIT,
@@ -5200,7 +5199,7 @@ async def agent_tools_route_patch(
         name : typing.Optional[CommonValidPythonIdentifier]
             Name of the tool (must be unique for this agent and a valid python identifier string )
 
-        function : typing.Optional[ToolsFunctionDefUpdate]
+        function : typing.Optional[ToolsFunctionDef]
 
         integration : typing.Optional[typing.Any]
 

sdks/python/julep/api/reference.md

@@ -2158,7 +2158,7 @@ client.agent_tools_route_patch(
 <dl>
 <dd>
 
-**function:** `typing.Optional[ToolsFunctionDefUpdate]` 
+**function:** `typing.Optional[ToolsFunctionDef]` 
 
 </dd>
 </dl>

sdks/python/julep/api/types/__init__.py

@@ -253,7 +253,6 @@
 from .tools_chosen_tool_call import ToolsChosenToolCall, ToolsChosenToolCall_Function
 from .tools_function_call_option import ToolsFunctionCallOption
 from .tools_function_def import ToolsFunctionDef
-from .tools_function_def_update import ToolsFunctionDefUpdate
 from .tools_function_tool import ToolsFunctionTool
 from .tools_named_function_choice import ToolsNamedFunctionChoice
 from .tools_named_tool_choice import ToolsNamedToolChoice, ToolsNamedToolChoice_Function
@@ -467,7 +466,6 @@
     "ToolsChosenToolCall_Function",
     "ToolsFunctionCallOption",
     "ToolsFunctionDef",
-    "ToolsFunctionDefUpdate",
     "ToolsFunctionTool",
     "ToolsNamedFunctionChoice",
     "ToolsNamedToolChoice",

sdks/python/julep/api/types/tools_function_def.py

@@ -6,27 +6,24 @@
 from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 from .common_identifier_safe_unicode import CommonIdentifierSafeUnicode
-from .common_valid_python_identifier import CommonValidPythonIdentifier
 
 
 class ToolsFunctionDef(pydantic_v1.BaseModel):
     """
     Function definition
     """
 
-    name: typing.Optional[CommonValidPythonIdentifier] = pydantic_v1.Field(default=None)
-    """
-    DO NOT USE: This will be overriden by the tool name. Here only for compatibility reasons.
-    """
-
+    name: typing.Optional[typing.Any] = None
     description: typing.Optional[CommonIdentifierSafeUnicode] = pydantic_v1.Field(
         default=None
     )
     """
     Description of the function
     """
 
-    parameters: typing.Dict[str, typing.Any] = pydantic_v1.Field()
+    parameters: typing.Optional[typing.Dict[str, typing.Any]] = pydantic_v1.Field(
+        default=None
+    )
     """
     The parameters the function accepts
     """