Update

app.py

@@ -8,7 +8,7 @@ st.title("Talk with Mingzhe")
 
 llm_client = LLMClient(api_key=st.secrets["OPENAI_API_KEY"], model_name="gpt-4-turbo-preview")
 pinecone_client = PersonalIndexClient(index_token=st.secrets["PINECONE_API_KEY"], embedding_token=st.secrets["OPENAI_API_KEY"], embedding_model_name='text-embedding-3-large', index_name='mingzhe')
-web_searcher = WebSearcher(search_token=st.secrets["YOU_API_KEY"])
+web_searcher = WebSearcher(you_api_key=st.secrets["YOU_API_KEY"], bing_api_key=st.secrets["BING_API_KEY"])
 
 if "messages" not in st.session_state:
     st.session_state.messages = []
@@ -32,7 +32,8 @@ if prompt := st.chat_input("What's up?"):
         web_query = llm_client.web_query_generator(query=prompt, history=st.session_state.messages)
         print(f"Web Query: {web_query}")
         
-        web_result = web_searcher.query_web_llm(query=prompt, num_web_results=10) if web_query != 'None' else "None"
+        web_result = web_searcher.query_bing(query=prompt) if web_query != 'None' else "None"
+        # web_result = web_searcher.query_web_llm(query=prompt, num_web_results=10) if web_query != 'None' else "None"
         print(f"Web Result: {web_result}")
         
         stream = llm_client.response_generate(prompt, st.session_state.messages, memory, web_result)

components.py

@@ -133,12 +133,21 @@ class PersonalIndexClient(object):
         return pinecone_memory
         
 class WebSearcher(object):
-    def __init__(self, search_token) -> None:
-        self.search_token = search_token
+    def __init__(self, you_api_key, bing_api_key) -> None:
+        self.you_api_key = you_api_key
+        self.bing_api_key = bing_api_key
         pass
 
     def query_web_llm(self, query, num_web_results=5):
-        headers = {"X-API-Key": self.search_token}
+        headers = {"X-API-Key": self.you_api_key}
         params = {"query": query, 'num_web_results': num_web_results}
         response_json = requests.get(f"https://api.ydc-index.io/rag?query={query}", params=params, headers=headers).json()
-        return response_json
+        return response_json
+    
+    def query_bing(self, query):
+        headers = {"Ocp-Apim-Subscription-Key": self.bing_api_key}
+        params = {"q": query, "textDecorations": True, "textFormat": "HTML"}
+        response = requests.get("https://api.bing.microsoft.com/v7.0/search", headers=headers, params=params)
+        response.raise_for_status()
+        search_results = response.json()
+        return search_results

playground.ipynb

@@ -9,48 +9,6 @@
     "import uuid"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "sid = uuid.uuid1()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 5,
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "'4d742a38dea811ee8aad1e30bc5366a5'"
-      ]
-     },
-     "execution_count": 5,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "sid.hex"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "pi.create(data=[{\n",
-    "        'id': '123',\n",
-    "        'content': '123',\n",
-    "        'metadata': {'hello': 'world'},\n",
-    "    }])"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": null,