Integrate Google Custom Search tool

app.py

@@ -13,6 +13,7 @@ from langchain_core.messages import AIMessage, HumanMessage
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain_core.pydantic_v1 import BaseModel, Field
 from langchain_google_genai import ChatGoogleGenerativeAI
+from google_custom_search import custom_google_search
 
 GOOGLE_API = "AIzaSyAz7e9gxDpUomG1YrE1W0evKC16cHvqgKc"
 
@@ -104,8 +105,8 @@ def main():
     agent_chain = create_agent_chain(tools, llm)
     user_input = get_user_input()
     if user_input:
-        response = llm.invoke(user_input)
-        display_response(response)
+        # response = llm.invoke(user_input)
+        # display_response(response)
         # prompt = """
         # You are a fact-checker. You are asked to verify the following statement based on the information you get from your tool, the search result we provided, 
         # and your knowledge. You should provide a response that is based on the information you have and that is as accurate as possible.
@@ -118,11 +119,10 @@ News Article Title:
 """
         prompt += f'"{response}"\n'
         prompt += "\n5 Strings from reputable news sites:\n"
-        prompt += '"dummy text"'
-        prompt += '"dummy text"'
-        prompt += '"dummy text"'
-        prompt += '"dummy text"'
-        prompt += '"dummy text"'
+
+        customSearchResults = custom_google_search(response)
+        for result in customSearchResults:
+            prompt += result
 
         new_prompt = st.text_area(prompt)
 

google_custom_search.py

@@ -0,0 +1,54 @@
+# This file/function uses a custom implemented Google Custom Search tool to search and return the top 5 news article titles for a given prompt
+
+from googleapiclient.discovery import build
+import pprint
+
+# Defaults to using Dustin's GCP API keys
+def custom_google_search(search_term, api_key="AIzaSyA4oDDFtPxAfmPC8EcfQrkByb9xKm2QfMc", cse_id="31e85635d41bd4040", num_results=5):
+    """Performs a Google Custom Search and returns titles of top results.
+
+    Args:
+        search_term: The search query.
+        api_key: Your Google API Key.
+        cse_id: Your Google Custom Search Engine ID.
+        num_results: The number of desired results (default: 5).
+
+    Returns:
+        A list of titles for the top search results.
+    """
+
+    service = build("customsearch", "v1", developerKey=api_key)
+    results = service.cse().list(
+        q=search_term,
+        cx=cse_id,
+        num=num_results  # Adjust the number of results as needed
+    ).execute()
+
+    search_result_titles = []
+    for item in results['items']:
+      # extended_title = item['title'] + item['snippet']
+      try:
+        # extended_title = item['pagemap']['metatags'][0]['og:description']
+        extended_title = "* '" + item['pagemap']['metatags'][0]['og:description'] + "'"
+      except:
+        # extended_title = item['title']
+        extended_title = "* '" + item['title'] + "'"
+
+      # pprint.pprint(f"dah=={item['pagemap']['metatags'][0]['og:description']}")
+      search_result_titles.append(extended_title)
+    # search_result_titles = [item['title'] for item in results['items']]
+    return search_result_titles
+
+if __name__ == "__main__":
+    # Examples
+    # search_phrase = "Californias governor rejects a bill to give unemployment checks to striking workers" # True
+    # search_phrase = "1,000 US troops deploying to build offshore port for Gaza aid" # True
+    # search_phrase = "More than 2500 migrants crossing the Mediterranean died or went missing this year" # True
+    # search_phrase = "Everything JK Rowling Would Be Willing To Do To Protect Her AntiTrans Views" # False
+    search_phrase = "Pat McAfee thought USC coach Lincoln Riley died because of President Biden's State of the Union address" # Unknown
+    top_titles = custom_google_search(search_phrase, my_api_key, my_cse_id)
+
+    print("Top Search Results:")
+    print("===================")
+    for title in top_titles:
+        print(f'{title}')