Update webscout.py

webscout.py

@@ -12,6 +12,13 @@ from typing import Dict, List, Optional, Tuple, Type, Union, cast
 import asyncio
 import json
 import aiohttp
+import requests
+import http.cookiejar as cookiejar
+import json
+from xml.etree import ElementTree
+import re
+import html.parser
+from typing import List, Dict, Union, Optional
 
 import pyreqwest_impersonate as pri
 
@@ -178,6 +185,7 @@ class WEBS:
             verify=False,
         )
         self._exception_event = Event()
+        self._chat_messages: List[Dict[str, str]] = []
         self._chat_vqd: str = ""
 
     def __enter__(self) -> "WEBS":
@@ -226,7 +234,7 @@ class WEBS:
         resp_content = self._get_url("POST", "https://duckduckgo.com", data={"q": keywords})
         return _extract_vqd(resp_content, keywords)
 
-    def chat(self, chat_messages: List[Dict[str, str]] = [], model: str = "gpt-3.5", timeout: int = 20) -> str:
+    def chat(self, keywords: str, model: str = "gpt-3.5", timeout: int = 20) -> str:
         """Initiates a chat session with DuckDuckGo AI.
 
         Args:
@@ -241,7 +249,7 @@ class WEBS:
         models = {
             "claude-3-haiku": "claude-3-haiku-20240307",
             "gpt-3.5": "gpt-3.5-turbo-0125",
-            "llama-3-70b": "meta-llama/Llama-3-70b-chat-hf",
+            "llama-3-70b": "meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo",
             "mixtral-8x7b": "mistralai/Mixtral-8x7B-Instruct-v0.1",
             "gpt-4o-mini": "gpt-4o-mini",
         }
@@ -250,9 +258,11 @@ class WEBS:
             resp = self.client.get("https://duckduckgo.com/duckchat/v1/status", headers={"x-vqd-accept": "1"})
             self._chat_vqd = resp.headers.get("x-vqd-4", "")
 
+        self._chat_messages.append({"role": "user", "content": keywords})
+
         json_data = {
             "model": models[model],
-            "messages": chat_messages,
+            "messages": self._chat_messages,
         }
         resp = self.client.post(
             "https://duckduckgo.com/duckchat/v1/chat",
@@ -264,6 +274,8 @@ class WEBS:
 
         data = ",".join(x for line in resp.text.rstrip("[DONE]\n").split("data:") if (x := line.strip()))
         result = "".join(x.get("message", "") for x in json_loads("[" + data + "]"))
+
+        self._chat_messages.append({"role": "assistant", "content": result})
         return result
 
     def text(
@@ -1169,15 +1181,8 @@ class WEBS:
             raise e
 
         return results
-import requests
-import http.cookiejar as cookiejar
-import json
-from xml.etree import ElementTree
-import re
-import html.parser
-from typing import List, Dict, Union, Optional
 
-html_parser = html.parser.HTMLParser()
+html_parser = html.parser.HTMLParser()html_parser = html.parser.HTMLParser()
 
 
 def unescape(string):