llm log (#3414)

skyvern/services/script_service.py

@@ -1360,6 +1360,7 @@ async def generate_text(
             json_response = await app.SINGLE_INPUT_AGENT_LLM_API_HANDLER(
                 prompt=script_generation_input_text_prompt,
                 prompt_name="script-generation-input-text-generatiion",
+                organization_id=context.organization_id,
             )
             new_text = json_response.get("answer", new_text)
         except Exception:

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

skyvern/services/task_v2_service.py

@@ -1284,6 +1284,7 @@ async def _generate_extraction_task(
         generate_extraction_task_prompt,
         task_v2=task_v2,
         prompt_name="task_v2_generate_extraction_task",
+        organization_id=task_v2.organization_id,
     )
     LOG.info("Data extraction response", data_extraction_response=generate_extraction_task_response)