vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-14 09:19:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Add step / task / workflow run / observer metrics as logs (#1698)

Browse source 
Co-authored-by: Suchintan <suchintan@users.noreply.github.com>
This commit is contained in:
Shuchang Zheng 2025-02-02 03:10:38 +08:00 • committed by GitHub
parent 41e8d8b0ac
commit 204972e225
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
11 changed files with 284 additions and 219 deletions
Download patch file Download diff file Expand all files Collapse all files

8
skyvern/forge/sdk/workflow/models/block.py
View file

@ -246,7 +246,9 @@ class Block(BaseModel, abc.ABC):
"generate_workflow_run_block_description",
block=block_data,
)
json_response = await app.SECONDARY_LLM_API_HANDLER(prompt=description_generation_prompt)
json_response = await app.SECONDARY_LLM_API_HANDLER(
prompt=description_generation_prompt, prompt_name="generate-workflow-run-block-description"
)
description = json_response.get("summary")
LOG.info(
"Generated description for the workflow run block",
@ -1198,7 +1200,7 @@ class TextPromptBlock(Block):
prompt=prompt,
llm_key=self.llm_key,
)
response = await llm_api_handler(prompt=prompt)
response = await llm_api_handler(prompt=prompt, prompt_name="text-prompt")
LOG.info("TextPromptBlock: Received response from LLM", response=response)
return response
@ -1954,7 +1956,7 @@ class PDFParserBlock(Block):
llm_prompt = prompt_engine.load_prompt(
"extract-information-from-file-text", extracted_text_content=extracted_text, json_schema=self.json_schema
)
llm_response = await app.LLM_API_HANDLER(prompt=llm_prompt)
llm_response = await app.LLM_API_HANDLER(prompt=llm_prompt, prompt_name="extract-information-from-file-text")
# Record the parsed data
await self.record_output_parameter_value(workflow_run_context, workflow_run_id, llm_response)
return await self.build_block_result(