ds-fire-fighter/test.py

import os
import requests
import json
from typing import List
from dotenv import load_dotenv
load_dotenv()


from src.llm import ai_chat


#conversation_id = "12345"  # Replace with the actual conversation ID
query = "Hello let us continue"
theme_id = 1
resume = "Emergency Response Specialist"
conversation_id = 1
response = ai_chat(query, conversation_id, theme_id, resume)
print(response)
"""
with open(file_path, 'rb') as file:
    files = {'file': file}
    response = requests.post(upload_url, files=files)
    response.raise_for_status()  # Ensure we raise an error for bad responses
    response_data = response.json()  # Get the response in JSON format
    print(response_data)

"""