Modifying Config Files

Agents/rankerAgent.py

@@ -1,14 +1,14 @@
 import json
+import os
 from together import Together
 
 def rerank_best_answer(json_files, config_file='config.json', model="meta-llama/Llama-3-8b-chat-hf"):
     # Load API key from configuration file
-    with open(config_file, 'r') as file:
-        config = json.load(file)
 
-    together_ai_key = config.get("TOGETHER_AI")
+
+    together_ai_key = os.getenv("TOGETHER_AI")
     if not together_ai_key:
-        raise ValueError("TOGETHER_AI key not found in the config file.")
+        raise ValueError("TOGETHER_AI environment variable not found. Please set it before running the script.")
 
     # Initialize Together client
     client = Together(api_key=together_ai_key)
@@ -81,12 +81,10 @@ Just Output the best_answer from above JSON and nothing else.
 
 def rankerAgent(prompt, config_file='config.json', model="meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo"):
     # Load API key from configuration file
-    with open(config_file, 'r') as file:
-        config = json.load(file)
-
-    together_ai_key = config.get("TOGETHER_AI")
+    together_ai_key = os.getenv("TOGETHER_AI")
     if not together_ai_key:
-        raise ValueError("TOGETHER_AI key not found in the config file.")
+        raise ValueError("TOGETHER_AI environment variable not found. Please set it before running the script.")
+
 
     # Initialize Together client
     client = Together(api_key=together_ai_key)
@@ -131,28 +129,3 @@ Just Output the best_answer from above JSON and nothing else.
         
     return response_bestModel.choices[0].message.content, response_bestAnswer.choices[0].message.content
     
-
-# # Usage example
-# json_files = ["../QnA_Eval/Responses/BOW_1_2_top_100_response.json",
-#               "../QnA_Eval/Responses/BOW_1_2_top_100_modified_response.json",
-#               "../QnA_Eval/Responses/tf-idf_1_2_top_100_response.json", 
-#               "../QnA_Eval/Responses/tf-idf_1_2_top_100_modified_response.json",
-#               "../QnA_Eval/Responses/bm25_1_2_top_100_response.json",
-#               "../QnA_Eval/Responses/bm25_1_2_top_100_modified_response.json",
-#               "../QnA_Eval/Responses/open_source_1_2_top_100_response.json", 
-#               "../QnA_Eval/Responses/open_source_1_2_top_100_modified_response.json",
-#               "../QnA_Eval/Responses/vision_1_2_top_100_response.json", 
-#               "../QnA_Eval/Responses/vision_1_2_top_100_modified_response.json",
-#               "../QnA_Eval/Responses/ZeroShot_response.json", 
-#               "../QnA_Eval/Responses/WikiAgent_response.json",
-#               "../QnA_Eval/Responses/WikiAgent_response_modified.json",
-#               "../QnA_Eval/Responses/LlamaAgent_response.json",
-#               "../QnA_Eval/Responses/LlamaAgent_response_modified.json",
-#               "../QnA_Eval/Responses/tf_idf_bm25_open_1_2_top_100_combined_response.json", "../QnA_Eval/Responses/tf_idf_bm25_open_1_2_top_100_combined_modified_response.json", "../QnA_Eval/Responses/tf_idf_bm25_open_1_2_top_100_combined_both_response.json"]
-
-# config_file = "../config.json"
-
-# result = rerank_best_answer(json_files, config_file)
-
-# with open("reranked_best_answers_1_2.json", 'w') as file:
-#     json.dump(result, file, indent=4, ensure_ascii=False)

Agents/togetherAIAgent.py

@@ -1,4 +1,4 @@
-import json
+import json, os
 from together import Together
 
 def generate_article_from_query(query, config_file='config.json', model="meta-llama/Llama-3-8b-chat-hf"):
@@ -13,13 +13,11 @@ def generate_article_from_query(query, config_file='config.json', model="meta-ll
     Returns:
     - str: The generated article content.
     """
-    # Load API key from configuration file
-    with open(config_file, 'r') as file:
-        config = json.load(file)
 
-    together_ai_key = config.get("TOGETHER_AI")
+    together_ai_key = os.getenv("TOGETHER_AI")
     if not together_ai_key:
-        raise ValueError("TOGETHER_AI key not found in the config file.")
+        raise ValueError("TOGETHER_AI environment variable not found. Please set it before running the script.")
+
 
     # Initialize Together client
     client = Together(api_key=together_ai_key)

AnswerGeneration/getAnswer.py

@@ -3,10 +3,10 @@ import json
 from together import Together
 
 def generate_answer_withContext(question, context):
-    with open('config.json', 'r') as file:
-        config = json.load(file)
+    together_ai_key = os.getenv("TOGETHER_AI")
+    if not together_ai_key:
+        raise ValueError("TOGETHER_AI environment variable not found. Please set it before running the script.")
 
-    together_ai_key = config.get("TOGETHER_AI")
 
     client = Together(api_key=together_ai_key) 
 
@@ -26,10 +26,10 @@ Question: {question}
 
 
 def generate_answer_zeroShot(question):
-    with open('config.json', 'r') as file:
-        config = json.load(file)
+    together_ai_key = os.getenv("TOGETHER_AI")
+    if not together_ai_key:
+        raise ValueError("TOGETHER_AI environment variable not found. Please set it before running the script.")
 
-    together_ai_key = config.get("TOGETHER_AI")
 
     client = Together(api_key=together_ai_key)
 

Query_Modification/QueryModification.py

@@ -1,14 +1,13 @@
-import json
+import json, os
 import pandas as pd
 import google.generativeai as genai
 
 # Function to process text input with Gemini model
 def query_Modifier(input_text):
 
-    with open('config.json', 'r') as file:
-        config = json.load(file)
-
-    gemini_key = config.get("GEMINI")
+    gemini_key = os.getenv("GEMINI")
+    if not gemini_key:
+        raise ValueError("GEMINI environment variable not found. Please set it before running the script.")
 
     # Initialize the API key
     genai.configure(api_key=gemini_key)
@@ -65,10 +64,9 @@ def query_Modifier(input_text):
 def getKeywords(input_text):
     # Extract keywords from the input text
 
-    with open('config.json', 'r') as file:
-        config = json.load(file)
-
-    gemini_key = config.get("GEMINI")
+    gemini_key = os.getenv("GEMINI")
+    if not gemini_key:
+        raise ValueError("GEMINI environment variable not found. Please set it before running the script.")
 
     # Initialize the API key
     genai.configure(api_key=gemini_key)