Now using Dependency Injection to associate conversations with requests (#8863)

openhands/server/utils.py

@@ -1,7 +1,7 @@
-from fastapi import Request
+from fastapi import Depends, Request
 
-from openhands.server.shared import ConversationStoreImpl, config
-from openhands.server.user_auth import get_user_auth
+from openhands.server.shared import ConversationStoreImpl, config, conversation_manager
+from openhands.server.user_auth import get_user_auth, get_user_id
 from openhands.storage.conversation.conversation_store import ConversationStore
 
 
@@ -16,3 +16,16 @@ async def get_conversation_store(request: Request) -> ConversationStore | None:
     conversation_store = await ConversationStoreImpl.get_instance(config, user_id)
     request.state.conversation_store = conversation_store
     return conversation_store
+
+
+async def get_conversation(
+    conversation_id: str, user_id: str | None = Depends(get_user_id)
+):
+    """Grabs conversation id set by middleware. Adds the conversation_id to the openapi schema."""
+    conversation = await conversation_manager.attach_to_conversation(
+        conversation_id, user_id
+    )
+    try:
+        yield conversation
+    finally:
+        await conversation_manager.detach_from_conversation(conversation)