Next commit

src/genai/analytics_chatbot/handlers/compare.py

@@ -0,0 +1,26 @@
+
+import requests
+from langchain_core.messages import SystemMessage,HumanMessage
+from ..utils.schemas import CompareBodyFormatter
+from ..utils.prompts import get_body_prompt
+from ..utils.utils import process_query
+
+def compare(state,llm_gpt,url):
+    messages = [SystemMessage(content=get_body_prompt()), 
+    HumanMessage(content=str(state['messages']))]
+    response=llm_gpt.with_structured_output(CompareBodyFormatter , method='function_calling').invoke(messages)
+    print('INF names response:', response)
+    payload = {
+    "usernames": list(map(process_query,response.names)),
+    "freq": response.frequency          
+    }
+    
+    print('The payload is:',payload)
+    
+    headers = {
+    "Content-Type": "application/json"
+    }
+
+    response = requests.post(url, json=payload, headers=headers)
+    print('Data from api:', response)
+    return response

src/genai/analytics_chatbot/utils/__init__.py

@@ -0,0 +1 @@
+init_variable = 5

src/genai/analytics_chatbot/utils/nodes.py

@@ -2,13 +2,14 @@ import requests
 from langchain_core.messages import SystemMessage , HumanMessage , FunctionMessage
 from .state import State
 from .tools import RetrieverBackup
-from .schemas import ResponseFormatter , CompareBodyFormatter, ParameterFormatter, EndpointFormatter
+from .schemas import CompareBodyFormatter, ParameterFormatter, EndpointFormatter
 from .prompts import query_check_prompt , get_body_prompt , fetch_last_message_prompt , fetch_parameters_prompt, fetch_endpoint_prompt, backup_retrieval_prompt
-from .utils import generate_api_knowledge , process_query, get_endpoint_info
+from .utils import  process_query, get_endpoint_info
 from src.genai.utils.models_loader import llm_gpt
 import numpy as np
 from src.genai.utils.data_loader import api_knowledge_df, api_index, caption_df , caption_index
 from src.genai.utils.models_loader import embedding_model
+from ..handlers.compare import compare
 
 
 class FetchLastMessage:
@@ -135,27 +136,31 @@ class FetchDataNode:
             url = f'''{self.base_url}{state['endpoint']}'''
 
             if state['endpoint'] == '/api/v1/compare/':
-                print('Condition satisfied')
-                messages = [SystemMessage(content=get_body_prompt()), 
-                HumanMessage(content=str(state['messages']))]
-                response=llm_gpt.with_structured_output(CompareBodyFormatter , method='function_calling').invoke(messages)
-                print('INF names response:', response)
-                payload = {
-                "usernames": list(map(process_query,response.names)),
-                "freq": response.frequency          
-                }
+                response=compare(state,llm_gpt,url)
+                return {'response': response.json()}
+                # print('Condition satisfied')
+                # messages = [SystemMessage(content=get_body_prompt()), 
+                # HumanMessage(content=str(state['messages']))]
+                # response=llm_gpt.with_structured_output(CompareBodyFormatter , method='function_calling').invoke(messages)
+                # print('INF names response:', response)
+                # payload = {
+                # "usernames": list(map(process_query,response.names)),
+                # "freq": response.frequency          
+                # }
                 
-                print('The payload is:',payload)
+                # print('The payload is:',payload)
                 
-                headers = {
-                "Content-Type": "application/json"
-                }
+                # headers = {
+                # "Content-Type": "application/json"
+                # }
 
-                response = requests.post(url, json=payload, headers=headers)
-                return {'response': response.json()}
+                # response = requests.post(url, json=payload, headers=headers)
+                # print('Data from api:', response)
+                # return {'response': response.json()}
             
             elif 'single_influencer_query' in  state['query_type']:
                 response = requests.get(url, params=state['parameters_values'],headers=self.headers)
+                print('Data from api:', response)
                 return {'response':response.json()}
             
             elif 'aggregate_query' in state['query_type']:
@@ -172,10 +177,9 @@ class FetchDataNode:
 
                         response = requests.get(url, params=current_params, headers=self.headers)
                         results[username] = response.json()  # Store influencer-wise response
-
+                    print('Data from api:', response)
                     return {"response": results}
             
-            print('Data from api:', response)
 
         except Exception as e:
             print('Error occoured:', e)