diff --git a/evaluation/auto_evaluation/eval_main.py b/evaluation/auto_evaluation/eval_main.py index c570282b..7b79bd44 100644 --- a/evaluation/auto_evaluation/eval_main.py +++ b/evaluation/auto_evaluation/eval_main.py @@ -26,11 +26,8 @@ # List of all available retrievers ALL_RETRIEVERS = { - "agent-retriever": "/graphs/agent-retriever", - "agent-retriever-reranker": "/graphs/agent-retriever", - "hybrid": "/graphs/hybrid", - "sim": "/graphs/sim", - "ensemble": "/graphs/ensemble", + "agent-retriever": "/conversations/agent-retriever", + "agent-retriever-reranker": "/conversations/agent-retriever", } RETRY_INTERVAL = 5 RETRY_TIMEOUT = 600 diff --git a/evaluation/script_based_evaluation/utils/api_utils.py b/evaluation/script_based_evaluation/utils/api_utils.py index bb54836f..9fea0dc9 100644 --- a/evaluation/script_based_evaluation/utils/api_utils.py +++ b/evaluation/script_based_evaluation/utils/api_utils.py @@ -17,7 +17,7 @@ def send_request( try: print("Sending request to endpoint:", endpoint) if endpoint in agent_retriever_urls: - url = f"{agent_retriever_urls[endpoint]}/graphs/agent-retriever" + url = f"{agent_retriever_urls[endpoint]}/conversations/agent-retriever" elif endpoint == "base-gemini-1.5-flash": response_text, response_time = base_gemini_1_5_flash(query) print("Response:", response_text) diff --git a/frontend/streamlit_app.py b/frontend/streamlit_app.py index d652bc8d..26b9eccd 100644 --- a/frontend/streamlit_app.py +++ b/frontend/streamlit_app.py @@ -109,7 +109,7 @@ def main(): st.title("OR Assistant") base_url = os.getenv("CHAT_ENDPOINT", "http://localhost:8000") - selected_endpoint = "/graphs/agent-retriever" + selected_endpoint = "/conversations/agent-retriever" if "selected_endpoint" not in st.session_state: st.session_state.selected_endpoint = selected_endpoint diff --git a/frontend/utils/mock_endpoint.py b/frontend/utils/mock_endpoint.py index 5d121c92..2a8eb5bd 100644 --- a/frontend/utils/mock_endpoint.py +++ b/frontend/utils/mock_endpoint.py @@ -33,7 +33,7 @@ def list_all_chains() -> List[str]: return ["/chains/mock"] -@app.post("/graphs/agent-retriever", response_model=ChatResponse) +@app.post("/conversations/agent-retriever", response_model=ChatResponse) def chat_app(request: ChatRequest) -> ChatResponse: """ Endpoint to handle chat requests.