finalizing v0.0.2

Author: kspviswa

prompt.py

@@ -27,7 +27,7 @@ def firePrompt(prompt: str, model: str="dolphin-mistral:latest", temp=0.4, isMlx
             if isMlx:
                 res = f'Search not implemented for MlX models. Stay tuned!'
             else:
-                res, keywords = retSearchResults(model=model, search_str=prompt)
+                res, keywords = retSearchResults(model=model, search_str=prompt, temp=temp)
         except Exception as e:
             return f'__Generic Error Occured__ {e}', ""
     return res, keywords

prompts.txt

@@ -1,5 +1,5 @@
 system_prompt = f"""
-    You are a helpful assistant capable of chatting with user and also perform a web research if required. Answer all questions truthfully to the best you can. If and only if you cannot answer the user_prompt, then use the information available in search_text to summarize an answer for the the user_prompt. While generating the summary, cite the source from the given search_text by converting the citation as a clickable text using markdown format.
+    You are a helpful assistant capable of chatting with user and also perform a web research if required. Answer all questions truthfully to the best you can. If and only if you cannot answer the user_prompt, then use the information available in search_text to summarize an answer for the the user_prompt. While generating the summary, always cite the source from the given search_text by converting the citation as a clickable text using markdown format.
 
     search_text : 
     user_prompt :

roadmap.md

@@ -6,7 +6,7 @@
 search = DuckDuckGoSearchResults(api_wrapper=wrapper)
 
 search_prompt = f"""
-    You are a helpful assistant capable of performing a web research. If the user_prompt is not an instruction, task or question, donot proceed to summarize. Instead, politely request the user that you cannot engage in general chat. If not, Use the information available in search_text to summarize an answer for the the user_prompt. While generating the summary, cite the sources from the search_text by converting the citation as a hyperlink using markdown format. 
+    You are a helpful assistant capable of performing a web research. If the user_prompt is not an instruction, task or question, donot proceed to summarize. Instead, politely request the user that you cannot engage in general chat. If not, Use the information available in search_text to summarize an answer for the the user_prompt. While generating the summary, always cite the sources from the search_text by converting the citation as a hyperlink using markdown format. 
 
     search_text : 
     user_prompt :
@@ -25,11 +25,11 @@ def retPrompt(res, up):
     """
     return tprompt
 
-def retSearchResults(model: str = "", search_str: str = "") -> str:
-    keywords_llm = Ollama(model=model, system=keyword_prompt)
+def retSearchResults(model: str = "", search_str: str = "", temp=0.4) -> str:
+    keywords_llm = Ollama(model=model, system=keyword_prompt, temperature=temp)
     query_str = keywords_llm.invoke(f'input_text : \n {search_str}')
     print(f'Keywords for this search texts are {query_str}')
-    search_llm = Ollama(model=model, system=search_prompt)
+    search_llm = Ollama(model=model, system=search_prompt, temperature=temp)
     searchResults = search.run(query_str)
-    response = search_llm.invoke(retPrompt(searchResults, search_str))
+    response = search_llm.invoke(retPrompt(searchResults, search_str), temperature=temp)
     return response, query_str