P1idZddlZddlZddlmZmZmZmZmZddl m Z ddl m Z ddl mZejeZGdd eZy) z4 OpenAI API service for accessing LLM capabilities. N)DictListOptionalAnyUnion)settings)OpenAI) LLMServicec eZdZdZddeedeeffd Z ddedeededed ed ef d Z d ed e fd Z d e eeffdZ xZS) OpenAIServicez6 Service for interacting with the OpenAI API. api_keymodelc|xstj}|xstj}t|||t |j |_y)z Initialize the OpenAI service with API key and model. Args: api_key: OpenAI API key, defaults to settings.OPENAI_API_KEY model: OpenAI model to use, defaults to settings.OPENAI_MODEL )rN)rOPENAI_API_KEY OPENAI_MODELsuper__init__r rclient)selfrr __class__s 4/home/cursorai/projects/iching/ai/services/openai.pyrzOpenAIService.__init__sD4X44... %(T\\2 prompt system_prompt temperature max_tokenstop_preturncg}|r|jd|d|jd|d dg}dg}dg} |j|d} |j| vr|| d<|j|vr|| d<|j|vr|| d<n|| d <|jjjj d i| } | j d jjS#t$r'} tjd t| d } ~ wwxYw)a Get a text completion from the OpenAI API. Args: prompt: The user prompt to send to the model system_prompt: Optional system prompt to set context temperature: Controls randomness (0-1), lower is more deterministic max_tokens: Maximum number of tokens to generate top_p: Nucleus sampling parameter Returns: The text response from the model system)rolecontentuserzgpt-5)rmessagesrrmax_completion_tokensrrzError calling OpenAI API: N) appendrrchat completionscreatechoicesmessager# Exceptionloggererrorstr) rrrrrrr%&models_requiring_max_completion_tokensmodels_with_fixed_temperaturemodels_not_supporting_top_p api_paramsresponsees rget_completionzOpenAIService.get_completions)&  OOX-H I F;<# 7>Y 2.5I ),3) '$J zz!<<&+ 7#zz!>>,7 =)zzCC6@ 23+5 <(:t{{''33::HZHH##A&..66 6  LL5c!fX> ?  sB)C D!"DD model_keycvttdi}||vr |||_ytj d|dy)z Change the model being used by the service. Args: model_key: Key from the OPENAI_MODELS dictionary in settings Returns: True if successful, False otherwise OPENAI_MODELSTz Model key 'z%' not found in settings.OPENAI_MODELSF)getattrrrr/warning)rr9modelss r change_modelzOpenAIService.change_model_sB?B7   *DJ NN[ 3XY Zrc$ttdiS)z Get a dictionary of available models. Returns: Dictionary with model keys and their corresponding model IDs r;)r<r)rs rget_available_modelsz"OpenAIService.get_available_modelsqsx"55r)NN)Ngffffff?ig?)__name__ __module__ __qualname____doc__rr1rfloatintr8boolr?rrA __classcell__)rs@rr r s 3 3Xc] 37;,/)-&) ?"?&.sm?%*?$' ? $ ? /2 ?Bcd$6d38n6rr )rEosloggingtypingrrrrr django.confropenair llm_servicer getLoggerrBr/r r'rrrQsA 33 #   8 $k6Jk6r