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/services/task_v1_service.py
+++ b/skyvern/services/task_v1_service.py
@@ -47,7 +47,9 @@ async def generate_task(user_prompt: str, organization: Organization) -> TaskGen

    llm_prompt = prompt_engine.load_prompt("generate-task", user_prompt=user_prompt)
    try:
-        llm_response = await app.LLM_API_HANDLER(prompt=llm_prompt, prompt_name="generate-task")
+        llm_response = await app.LLM_API_HANDLER(
+            prompt=llm_prompt, prompt_name="generate-task", organization_id=organization.organization_id
+        )
        parsed_task_generation_obj = TaskGenerationBase.model_validate(llm_response)

        # generate a TaskGenerationModel