Update prompt_templates.py

prompt_templates.py

@@ -1,70 +1,70 @@
-from imports import *
-from prompts import *
-
-
-os.environ['HUGGINGFACEHUB_API_TOKEN'] = "<API>"
-os.environ['OPENAI_API_KEY'] = "<API>"
-# ENV_HOST = "https://cloud.langfuse.com"
-# ENV_SECRET_KEY = userdata.get('LF-SECRET')
-# ENV_PUBLIC_KEY = userdata.get('LF-PUBLIC')
-
-system_prompt_classifier = PromptTemplate(
-    input_variables=["QUERY", "PREV_QUERY"], template= system_prompt_classifier_template,
-    output_key="cassification"
-)
-
-system_memory_prompt = PromptTemplate(
-    input_variables=["QUERY", "API_LIST", "RAG", "PAST_QUERY", "PAST_RESPONSE"], template= system_prompt_memory_template
-)
-
-system_prompt = PromptTemplate(
-    input_variables=["QUERY", "API_LIST", "RAG"], template= system_prompt_template
-)
-
-follow_up_prompt = PromptTemplate(
-    input_variables=["QUERY", "chat_history"], template= follow_up_prompt_template
-)
-generation_prompt = PromptTemplate(
-    input_variables = ["API_LIST", "FEW_SHOT", "MODIFIED_ARG"],template = generation_prompt_template
-  )
-
-memory = ConversationBufferWindowMemory(
-    memory_key="chat_history", input_key = "QUERY", k = 1, 
-    return_messages=True
-)
-llm_gpt_3_5 = ChatOpenAI(temperature = 0.0, model =  "gpt-3.5-turbo-1106")
-llm_gpt_4 = ChatOpenAI(temperature = 0.0, model =  "gpt-4")
-
-mem_chain = LLMChain(llm=llm_gpt_3_5,
-                     prompt=system_prompt_classifier,
-                     memory = memory,
-                     verbose=True)
-
-query_chain_memory = LLMChain(llm=llm_gpt_3_5,
-                        prompt=system_memory_prompt,
-                        memory = memory,
-                        verbose=True)
-
-query_chain = LLMChain(llm=llm_gpt_3_5,
-                       prompt=system_prompt,
-                       memory = memory,
-                       verbose=True)
-
-format_chain = LLMChain(llm=llm_gpt_3_5,
-                        prompt=follow_up_prompt,
-                        memory = memory,
-                        verbose=True)
-
-generation_chain = LLMChain(llm= ChatOpenAI(temperature = 0.7, model =  "gpt-4"),
-                        prompt=generation_prompt,
-                        output_key = 'QUERY',
-                        verbose=False)
-reprompt = PromptTemplate(
-    input_variables=["QUERY", "API_LIST", "CORRECTION_PROMPT", "chat_history"], template= reprompt_template
-)
-
-reprompt_chain = LLMChain(llm=llm_gpt_4,
-                        prompt=reprompt,
-                        output_key = 'new_response',
-                        memory = memory,
-                        verbose=True)
+from imports import *
+from prompts import *
+
+
+os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.getenv('HF')
+os.environ['OPENAI_API_KEY'] = os.getenv('OpenAi')
+# ENV_HOST = "https://cloud.langfuse.com"
+# ENV_SECRET_KEY = userdata.get('LF-SECRET')
+# ENV_PUBLIC_KEY = userdata.get('LF-PUBLIC')
+
+system_prompt_classifier = PromptTemplate(
+    input_variables=["QUERY", "PREV_QUERY"], template= system_prompt_classifier_template,
+    output_key="cassification"
+)
+
+system_memory_prompt = PromptTemplate(
+    input_variables=["QUERY", "API_LIST", "RAG", "PAST_QUERY", "PAST_RESPONSE"], template= system_prompt_memory_template
+)
+
+system_prompt = PromptTemplate(
+    input_variables=["QUERY", "API_LIST", "RAG"], template= system_prompt_template
+)
+
+follow_up_prompt = PromptTemplate(
+    input_variables=["QUERY", "chat_history"], template= follow_up_prompt_template
+)
+generation_prompt = PromptTemplate(
+    input_variables = ["API_LIST", "FEW_SHOT", "MODIFIED_ARG"],template = generation_prompt_template
+  )
+
+memory = ConversationBufferWindowMemory(
+    memory_key="chat_history", input_key = "QUERY", k = 1, 
+    return_messages=True
+)
+llm_gpt_3_5 = ChatOpenAI(temperature = 0.0, model =  "gpt-3.5-turbo-1106")
+llm_gpt_4 = ChatOpenAI(temperature = 0.0, model =  "gpt-4")
+
+mem_chain = LLMChain(llm=llm_gpt_3_5,
+                     prompt=system_prompt_classifier,
+                     memory = memory,
+                     verbose=True)
+
+query_chain_memory = LLMChain(llm=llm_gpt_3_5,
+                        prompt=system_memory_prompt,
+                        memory = memory,
+                        verbose=True)
+
+query_chain = LLMChain(llm=llm_gpt_3_5,
+                       prompt=system_prompt,
+                       memory = memory,
+                       verbose=True)
+
+format_chain = LLMChain(llm=llm_gpt_3_5,
+                        prompt=follow_up_prompt,
+                        memory = memory,
+                        verbose=True)
+
+generation_chain = LLMChain(llm= ChatOpenAI(temperature = 0.7, model =  "gpt-4"),
+                        prompt=generation_prompt,
+                        output_key = 'QUERY',
+                        verbose=False)
+reprompt = PromptTemplate(
+    input_variables=["QUERY", "API_LIST", "CORRECTION_PROMPT", "chat_history"], template= reprompt_template
+)
+
+reprompt_chain = LLMChain(llm=llm_gpt_4,
+                        prompt=reprompt,
+                        output_key = 'new_response',
+                        memory = memory,
+                        verbose=True)