tweaks from presenting

Author: samuelcolvin

durable-exec/deep_research_dbos.py

@@ -1,5 +1,4 @@
 import asyncio
-import os
 import sys
 import uuid
 from typing import Annotated, List
@@ -114,7 +113,7 @@ async def deep_research_durable(query: str):
     config: DBOSConfig = {
         'name': 'deep_research_durable',
         'enable_otlp': True,
-        'conductor_key': os.environ.get('DBOS_CONDUCTOR_KEY', None),
+        'system_database_url': 'postgresql://postgres@localhost:5432/dbos',
     }
     DBOS(config=config)
     DBOS.launch()

durable-exec/twenty_questions_dbos.py

@@ -1,12 +1,12 @@
 import asyncio
-import os
 import sys
 import uuid
 from dataclasses import dataclass
 from enum import StrEnum
+from typing import Any
 
 import logfire
-from dbos import DBOS, DBOSConfig, SetWorkflowID
+from dbos import DBOS, DBOSConfig, SetWorkflowID, WorkflowHandle
 from pydantic_ai import Agent, AgentRunResult, RunContext, UsageLimits
 from pydantic_ai.durable_exec.dbos import DBOSAgent
 
@@ -78,20 +78,22 @@ async def play(resume_id: str | None, answer: str) -> AgentRunResult[str]:
     config: DBOSConfig = {
         'name': 'twenty_questions_durable',
         'enable_otlp': True,
-        'conductor_key': os.environ.get('DBOS_CONDUCTOR_KEY', None),
+        'system_database_url': 'postgresql://postgres@localhost:5432/dbos',
     }
     DBOS(config=config)
     DBOS.launch()
-    wf_id = f'twenty-questions-{uuid.uuid4()}'
+
     if resume_id is not None:
         print('resuming existing workflow', resume_id)
-        wf_id = resume_id
+        wf: WorkflowHandle[Any] = DBOS.retrieve_workflow(resume_id)
+        result = await wf.get_result()
     else:
+        wf_id = f'twenty-questions-{uuid.uuid4()}'
         print('starting new workflow', wf_id)
 
-    state = GameState(answer=answer)
-    with SetWorkflowID(wf_id):
-        result = await dbos_questioner_agent.run('start', deps=state, usage_limits=UsageLimits(request_limit=25))
+        state = GameState(answer=answer)
+        with SetWorkflowID(wf_id):
+            result = await dbos_questioner_agent.run('start', deps=state, usage_limits=UsageLimits(request_limit=25))
 
     print(f'After {len(result.all_messages()) / 2}, the answer is: {result.output}')
 

durable-exec/twenty_questions_temporal.py

@@ -34,7 +34,7 @@
 
 # Agent that asks questions to guess the object
 questioner_agent = Agent(
-    'anthropic:claude-sonnet-4-0',
+    'anthropic:claude-sonnet-4-5',
     instructions="""
 You are playing a question and answer game. You need to guess what object the other player is thinking of.
 Your job is to ask yes/no questions to narrow down the possibilities.
@@ -83,7 +83,7 @@ async def play(resume_id: str | None):
             print('resuming existing workflow', resume_id)
             await client.get_workflow_handle(resume_id).result()  # type: ignore[ReportUnknownMemberType]
         else:
-            await client.execute_workflow(  # type: ignore[ReportUnknownMemberType]
+            await client.execute_workflow(
                 TwentyQuestionsWorkflow.run,
                 id=f'twenty_questions-{uuid.uuid4()}',
                 task_queue='twenty_questions',

pyproject.toml

@@ -10,7 +10,7 @@ dependencies = [
     "fastapi>=0.115.14",
     "logfire[asyncpg,fastapi,httpx]>=4.10",
     "mcp>=1.15.0",
-    "pydantic-ai>=1.18.0",
+    "pydantic-ai>=1.20.0",
     "tavily-python>=0.7.12",
     "dbos>=2",
 ]