vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Change ai suggestion to take user input and context (#1620)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-01-23 03:01:12 +08:00
committed by GitHub
parent 8b55e1dad4
commit 6123b1d5f0
3 changed files with 49 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
skyvern/forge/sdk/routes/agent_protocol.py
View File

@@ -956,7 +956,7 @@ async def make_ai_suggestion(
llm_prompt = ""
if ai_suggestion_type == AISuggestionType.DATA_SCHEMA:
llm_prompt = prompt_engine.load_prompt("suggest-data-schema", input=data.input)
llm_prompt = prompt_engine.load_prompt("suggest-data-schema", input=data.input, additional_context=data.context)
try:
new_ai_suggestion = await app.DATABASE.create_ai_suggestion(

1
skyvern/forge/sdk/schemas/ai_suggestions.py
View File

@@ -20,3 +20,4 @@ class AISuggestion(AISuggestionBase):
class AISuggestionRequest(BaseModel):
input: str = Field(..., min_length=1)
context: dict[str, Any] | None = None