esg-rag-platform / api_calls.py
JiunYi's picture
update new feature
0d2b1ab
import requests
API_ENDPOINT = "http://54.254.230.28:8888"
# function to call api
def call_api_stream(api_path, api_params):
session = requests.Session()
url = f"{API_ENDPOINT}/{api_path}"
response = session.post(
url, json=api_params, headers={"Content-Type": "application/json"},
stream=True
)
return response
def call_api(api_path, api_params):
session = requests.Session()
url = f"{API_ENDPOINT}/{api_path}"
response = session.post(
url, json=api_params, headers={"Content-Type": "application/json"}
)
return response.json()
def api_rag_qa_chain_demo(openai_model_name, query, year, target_type, target_value, history):
# api_path = "qa/demo"
api_path = "qa/demo/test-feature"
api_params = {
"openai_model_name": openai_model_name,
"query": query,
"year": year,
"target_type": target_type,
"target_value": target_value,
"prev_turn_of_conversation": history,
}
return call_api_stream(api_path, api_params)
def api_rag_summ_chain_demo(openai_model_name, query, year, target_type, target_value, tone):
# api_path = "summary/demo"
api_path = "summary/demo/test-feature"
api_params = {
"openai_model_name": openai_model_name,
"query": query,
"year": year,
"target_type": target_type,
"target_value": target_value,
"tone": tone,
}
return call_api_stream(api_path, api_params)