More Type Safety (#10848)

This commit is contained in:
Tim O'Farrell
2025-09-05 11:34:43 -06:00
committed by GitHub
parent 3366ad9de7
commit aca568cfbe
4 changed files with 17 additions and 11 deletions

View File

@@ -45,7 +45,6 @@ from openhands.server.services.conversation_service import (
setup_init_conversation_settings,
)
from openhands.server.shared import (
ConversationManagerImpl,
ConversationStoreImpl,
config,
conversation_manager,
@@ -397,7 +396,7 @@ async def get_prompt(
)
prompt_template = generate_prompt_template(stringified_events)
prompt = generate_prompt(llm_config, prompt_template, conversation_id)
prompt = await generate_prompt(llm_config, prompt_template, conversation_id)
return JSONResponse(
{
@@ -413,7 +412,7 @@ def generate_prompt_template(events: str) -> str:
return template.render(events=events)
def generate_prompt(
async def generate_prompt(
llm_config: LLMConfig, prompt_template: str, conversation_id: str
) -> str:
messages = [
@@ -427,7 +426,7 @@ def generate_prompt(
},
]
raw_prompt = ConversationManagerImpl.request_llm_completion(
raw_prompt = await conversation_manager.request_llm_completion(
'remember_prompt', conversation_id, llm_config, messages
)
prompt = re.search(r'<update_prompt>(.*?)</update_prompt>', raw_prompt, re.DOTALL)