Function get_query_result

api/aifunctions/aiquery.py:49–87

(query: str)

def get_query_result(query: str) :
    if not is_open_ai_available():
        return {}
    messages = [
        {
            "role": "system",
            "content": f"""
                Configure query engine to handle RF recording conversions to Hertz frequencies.
                Prioritize polygon-based geolocation for accuracy.
                Use extended radii to capture all relevant recordings.
                Today's date is {datetime.now().strftime("%m/%d/%Y, %H:%M:%S")}.
            """
        },
        {
            "role": "user",
            "content": query
        }
    ]
    try:
        response = openai.ChatCompletion.create(
            engine=engine,
            messages = messages,
            functions = rf_function,
            temperature=0.2,
            max_tokens=800,
            top_p=0.95,
            frequency_penalty=0,
            presence_penalty=0,
            stop=None
        )
        response_message = response["choices"][0]["message"]
        if not response_message.get("function_call"):
            return {}
        function_args = json.loads(response_message["function_call"]["arguments"])
        return function_args

    except Exception as e:
        print(e)
        raise("Open AI is not available", e)