def call_openai_base1()

in Autogen_v0.4/rag_agent/openai_helper.py [0:0]

• 33 lines of code
• 5 McCabe index (conditional complexity)

def call_openai_base1(prompt, service_bus_session_id, queue = None):       
    
    try:
        messages = [prompt['gptPrompt']['systemMessage']] + [prompt['gptPrompt']['question']]
        response = openai.ChatCompletion.create(
                    engine=os.getenv('DEPLOYMENT_NAME'), 
                    messages = messages,
                    temperature= float(prompt['temperature']),
                    extra_body={
                        "dataSources": [
                            {
                                "type": "AzureCognitiveSearch",
                                "parameters": {
                                    "endpoint": os.environ["SEARCH_ENDPOINT"],
                                    "key": os.environ["SEARCH_KEY"],
                                    "indexName": os.environ["SEARCH_INDEX_NAME"],
                                }
                            }
                        ]
                    },
                    max_tokens=int(prompt['maxTokens']) ,
                    stream=True                    
                    )
        for chunk in response:
            try:
                content = chunk['choices'][0]['delta']['content']
                if content is not None:
                    queue.put(content)
            except (KeyError, IndexError, TypeError) as e:
                print(f"An error occurred: {e}")
           
    except Exception as e:
        print("Error calling OpenAI:" + openai.api_base)
        print(e)
        return str(e)