apps/stealth-ai-relay
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(bot): better buttons

Browse Source
This commit is contained in:
h
2026-01-21 03:00:22 +01:00
parent 5af751f575
commit 5441454993
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
backend/src/bot/modules/ai/agent.py
View File

@@ -41,7 +41,7 @@ def create_text_agent(
model = GoogleModel(model_name, provider=provider) model = GoogleModel(model_name, provider=provider)
base_prompt = system_prompt or DEFAULT_SYSTEM_PROMPT base_prompt = system_prompt or DEFAULT_SYSTEM_PROMPT
full_prompt = f"{base_prompt} {LATEX_INSTRUCTION}" full_prompt = f"{base_prompt} {LATEX_INSTRUCTION}"
return Agent(model, system_prompt=full_prompt) return Agent(model, instructions=full_prompt)
def create_follow_up_agent( def create_follow_up_agent(
@@ -52,7 +52,7 @@ def create_follow_up_agent(
provider = GoogleProvider(api_key=api_key) provider = GoogleProvider(api_key=api_key)
model = GoogleModel(model_name, provider=provider) model = GoogleModel(model_name, provider=provider)
prompt = system_prompt or DEFAULT_FOLLOW_UP prompt = system_prompt or DEFAULT_FOLLOW_UP
return Agent(model, output_type=FollowUpOptions, system_prompt=prompt) return Agent(model, output_type=FollowUpOptions, instructions=prompt)
def build_message_history(history: list[dict[str, str]]) -> list[ModelMessage]: def build_message_history(history: list[dict[str, str]]) -> list[ModelMessage]: