33from .basesdk import BaseSDK
44from openrouter import errors , models , utils
55from openrouter ._hooks import HookContext
6+ from openrouter .models import completioncreateparams as models_completioncreateparams
67from openrouter .types import OptionalNullable , UNSET
78from openrouter .utils import get_security_from_env
89from openrouter .utils .unmarshal_json_response import unmarshal_json_response
@@ -13,7 +14,10 @@ class Completions(BaseSDK):
1314 def generate (
1415 self ,
1516 * ,
16- prompt : Union [models .Prompt , models .PromptTypedDict ],
17+ prompt : Union [
18+ models_completioncreateparams .Prompt ,
19+ models_completioncreateparams .PromptTypedDict ,
20+ ],
1721 model : Optional [str ] = None ,
1822 models : Optional [List [str ]] = None ,
1923 best_of : OptionalNullable [int ] = UNSET ,
@@ -27,13 +31,16 @@ def generate(
2731 seed : OptionalNullable [int ] = UNSET ,
2832 stop : OptionalNullable [
2933 Union [
30- models .CompletionCreateParamsStop ,
31- models .CompletionCreateParamsStopTypedDict ,
34+ models_completioncreateparams .CompletionCreateParamsStop ,
35+ models_completioncreateparams .CompletionCreateParamsStopTypedDict ,
3236 ]
3337 ] = UNSET ,
3438 stream : Optional [bool ] = False ,
3539 stream_options : OptionalNullable [
36- Union [models .StreamOptions , models .StreamOptionsTypedDict ]
40+ Union [
41+ models_completioncreateparams .StreamOptions ,
42+ models_completioncreateparams .StreamOptionsTypedDict ,
43+ ]
3744 ] = UNSET ,
3845 suffix : OptionalNullable [str ] = UNSET ,
3946 temperature : OptionalNullable [float ] = UNSET ,
@@ -42,8 +49,8 @@ def generate(
4249 metadata : OptionalNullable [Dict [str , str ]] = UNSET ,
4350 response_format : OptionalNullable [
4451 Union [
45- models .CompletionCreateParamsResponseFormatUnion ,
46- models .CompletionCreateParamsResponseFormatUnionTypedDict ,
52+ models_completioncreateparams .CompletionCreateParamsResponseFormatUnion ,
53+ models_completioncreateparams .CompletionCreateParamsResponseFormatUnionTypedDict ,
4754 ]
4855 ] = UNSET ,
4956 retries : OptionalNullable [utils .RetryConfig ] = UNSET ,
@@ -187,7 +194,10 @@ def generate(
187194 async def generate_async (
188195 self ,
189196 * ,
190- prompt : Union [models .Prompt , models .PromptTypedDict ],
197+ prompt : Union [
198+ models_completioncreateparams .Prompt ,
199+ models_completioncreateparams .PromptTypedDict ,
200+ ],
191201 model : Optional [str ] = None ,
192202 models : Optional [List [str ]] = None ,
193203 best_of : OptionalNullable [int ] = UNSET ,
@@ -201,13 +211,16 @@ async def generate_async(
201211 seed : OptionalNullable [int ] = UNSET ,
202212 stop : OptionalNullable [
203213 Union [
204- models .CompletionCreateParamsStop ,
205- models .CompletionCreateParamsStopTypedDict ,
214+ models_completioncreateparams .CompletionCreateParamsStop ,
215+ models_completioncreateparams .CompletionCreateParamsStopTypedDict ,
206216 ]
207217 ] = UNSET ,
208218 stream : Optional [bool ] = False ,
209219 stream_options : OptionalNullable [
210- Union [models .StreamOptions , models .StreamOptionsTypedDict ]
220+ Union [
221+ models_completioncreateparams .StreamOptions ,
222+ models_completioncreateparams .StreamOptionsTypedDict ,
223+ ]
211224 ] = UNSET ,
212225 suffix : OptionalNullable [str ] = UNSET ,
213226 temperature : OptionalNullable [float ] = UNSET ,
@@ -216,8 +229,8 @@ async def generate_async(
216229 metadata : OptionalNullable [Dict [str , str ]] = UNSET ,
217230 response_format : OptionalNullable [
218231 Union [
219- models .CompletionCreateParamsResponseFormatUnion ,
220- models .CompletionCreateParamsResponseFormatUnionTypedDict ,
232+ models_completioncreateparams .CompletionCreateParamsResponseFormatUnion ,
233+ models_completioncreateparams .CompletionCreateParamsResponseFormatUnionTypedDict ,
221234 ]
222235 ] = UNSET ,
223236 retries : OptionalNullable [utils .RetryConfig ] = UNSET ,
0 commit comments