DataAgent/esg-rag-platform

1import requests2 3API_ENDPOINT = "http://54.254.230.28:8888"4 5# function to call api6def call_api_stream(api_path, api_params):7    session = requests.Session()8    url = f"{API_ENDPOINT}/{api_path}"9    response = session.post(10        url, json=api_params, headers={"Content-Type": "application/json"},11        stream=True12    )13    return response14 15def call_api(api_path, api_params):16    session = requests.Session()17    url = f"{API_ENDPOINT}/{api_path}"18    response = session.post(19        url, json=api_params, headers={"Content-Type": "application/json"}20    )21    return response.json()22 23def api_rag_qa_chain_demo(openai_model_name, query, year, company_name):24    api_path = "qa/demo"25    api_params = {26        "openai_model_name": openai_model_name,27        "query": query,28        "year": year,29        "company_name": company_name,30    }31    return call_api_stream(api_path, api_params)32 33def api_rag_summ_chain_demo(openai_model_name, query, year, company_name, tone):34    api_path = "summary/demo"35    api_params = {36        "openai_model_name": openai_model_name,37        "query": query,38        "year": year,39        "company_name": company_name,40        "tone": tone,41    }42    return call_api_stream(api_path, api_params)43