@@ -36,7 +36,7 @@ class while customizing it with any plugins.
3636 GenerateResponseWrapper ,
3737 ModelMiddleware ,
3838)
39- from genkit .blocks .prompt import to_generate_action_options
39+ from genkit .blocks .prompt import PromptConfig , to_generate_action_options
4040from genkit .core .action import ActionRunContext
4141from genkit .core .action .types import ActionKind
4242from genkit .types import (
@@ -161,24 +161,26 @@ async def generate(
161161 """
162162 return await generate_action (
163163 self .registry ,
164- to_generate_action_options (
165- registry = self .registry ,
166- model = model ,
167- prompt = prompt ,
168- system = system ,
169- messages = messages ,
170- tools = tools ,
171- return_tool_requests = return_tool_requests ,
172- tool_choice = tool_choice ,
173- tool_responses = tool_responses ,
174- config = config ,
175- max_turns = max_turns ,
176- output_format = output_format ,
177- output_content_type = output_content_type ,
178- output_instructions = output_instructions ,
179- output_schema = output_schema ,
180- output_constrained = output_constrained ,
181- docs = docs ,
164+ await to_generate_action_options (
165+ self .registry ,
166+ PromptConfig (
167+ model = model ,
168+ prompt = prompt ,
169+ system = system ,
170+ messages = messages ,
171+ tools = tools ,
172+ return_tool_requests = return_tool_requests ,
173+ tool_choice = tool_choice ,
174+ tool_responses = tool_responses ,
175+ config = config ,
176+ max_turns = max_turns ,
177+ output_format = output_format ,
178+ output_content_type = output_content_type ,
179+ output_instructions = output_instructions ,
180+ output_schema = output_schema ,
181+ output_constrained = output_constrained ,
182+ docs = docs ,
183+ )
182184 ),
183185 on_chunk = on_chunk ,
184186 middleware = use ,
@@ -289,7 +291,7 @@ def generate_stream(
289291 )
290292 stream .set_close_future (resp )
291293
292- return ( stream , stream .closed )
294+ return stream , stream .closed
293295
294296 async def embed (
295297 self ,
0 commit comments