def get_chatgpt_base_response()

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

• 17 lines of code
• 1 McCabe index (conditional complexity)

def get_chatgpt_base_response(userQuery, system_message , service_bus_session_id, queue=None):

    completion = client.chat.completions.create(
        model=os.getenv('DEPLOYMENT_NAME'),
        messages=[
            {
                "role": "system",
                "content": system_message
            },
            {
                "role": "user",
                "content": userQuery
            }
        ],
        temperature=0,
        max_tokens=100,
        stream=False)
    
    return completion.choices[0].message.content