policy_analyser/llm.py

"""
    Call OpenAI
    @author : Sakshi Tantak
"""

# Imports
from openai import AzureOpenAI

from policy_analyser import GPT_ENGINE, GPT_API_BASE, GPT_KEY, GPT_VERSION

CLIENT = AzureOpenAI(
  azure_endpoint = GPT_API_BASE,
  api_key = GPT_KEY,
  api_version = GPT_VERSION
)

def call_openai(system_prompt, seed = 42):
    print('Calling openai')
    # messages = [{'role' : 'system', 'content' : system_prompt},
                # {'role' : 'user', 'content' : document}]
    messages = [{'role' : 'user', 'content' : system_prompt}]
    response = CLIENT.chat.completions.create(
        model = GPT_ENGINE,
        messages = messages,
        # response_format = response_format,
        reasoning_effort = 'low'
    )
    print('LLM response : ', response)
    return response.choices[0].message.content