llm log (#3414)

2025-09-11 18:10:05 -07:00
parent 7a8974f6c7
commit 0e2aecc75d
11 changed files with 137 additions and 113 deletions
--- a/skyvern/forge/sdk/routes/agent_protocol.py
+++ b/skyvern/forge/sdk/routes/agent_protocol.py
@@ -1863,7 +1863,10 @@ async def suggest(
        )

        llm_response = await app.LLM_API_HANDLER(
-            prompt=llm_prompt, ai_suggestion=new_ai_suggestion, prompt_name="suggest-data-schema"
+            prompt=llm_prompt,
+            ai_suggestion=new_ai_suggestion,
+            prompt_name="suggest-data-schema",
+            organization_id=current_org.organization_id,
        )
        parsed_ai_suggestion = AISuggestionBase.model_validate(llm_response)