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

Don't allow None for org id (partial mypy only change) more to come (#2613)

Browse Source
This commit is contained in:
Asher Foa
2025-06-05 19:00:20 -04:00
committed by GitHub
parent 990b1c921f
commit e258c6332d
5 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
skyvern/forge/agent.py
View File

@@ -210,7 +210,7 @@ class ForgeAgent:
) )
return task, step return task, step
async def create_task(self, task_request: TaskRequest, organization_id: str | None = None) -> Task: async def create_task(self, task_request: TaskRequest, organization_id: str) -> Task:
webhook_callback_url = str(task_request.webhook_callback_url) if task_request.webhook_callback_url else None webhook_callback_url = str(task_request.webhook_callback_url) if task_request.webhook_callback_url else None
totp_verification_url = str(task_request.totp_verification_url) if task_request.totp_verification_url else None totp_verification_url = str(task_request.totp_verification_url) if task_request.totp_verification_url else None
task = await app.DATABASE.create_task( task = await app.DATABASE.create_task(

2
skyvern/forge/agent_functions.py
View File

@@ -463,7 +463,7 @@ class AgentFunction:
return return
async def validate_task_execution( async def validate_task_execution(
self, organization_id: str | None = None, task_id: str | None = None, task_version: str | None = None self, organization_id: str, task_id: str | None = None, task_version: str | None = None
) -> None: ) -> None:
return return

2
skyvern/forge/sdk/db/client.py
View File

@@ -312,7 +312,7 @@ class AgentDB:
LOG.error("UnexpectedError", exc_info=True) LOG.error("UnexpectedError", exc_info=True)
raise raise
async def get_task_steps(self, task_id: str, organization_id: str | None = None) -> list[Step]: async def get_task_steps(self, task_id: str, organization_id: str) -> list[Step]:
try: try:
async with self.Session() as session: async with self.Session() as session:
if steps := ( if steps := (

2
skyvern/forge/sdk/models.py
View File

@@ -49,7 +49,7 @@ class Step(BaseModel):
order: int order: int
is_last: bool is_last: bool
retry_index: int = 0 retry_index: int = 0
organization_id: str | None = None organization_id: str
input_token_count: int = 0 input_token_count: int = 0
output_token_count: int = 0 output_token_count: int = 0
reasoning_token_count: int | None = None reasoning_token_count: int | None = None

2
skyvern/forge/sdk/schemas/tasks.py
View File

@@ -231,7 +231,7 @@ class Task(TaskBase):
None, None,
description="The reason for the task failure.", description="The reason for the task failure.",
) )
organization_id: str | None = None organization_id: str
workflow_run_id: str | None = None workflow_run_id: str | None = None
workflow_permanent_id: str | None = None workflow_permanent_id: str | None = None
order: int | None = None order: int | None = None