Blacked.

autogpt/json_fixes/auto_fix.py

@@ -21,12 +21,14 @@ def fix_json(json_string: str, schema: str) -> str:
     # Try to fix the JSON using GPT:
     function_string = "def fix_json(json_string: str, schema:str=None) -> str:"
     args = [f"'''{json_string}'''", f"'''{schema}'''"]
-    description_string = "This function takes a JSON string and ensures that it"\
+    description_string = (
-        " is parseable and fully compliant with the provided schema. If an object"\
+        "This function takes a JSON string and ensures that it"
-        " or field specified in the schema isn't contained within the correct JSON,"\
+        " is parseable and fully compliant with the provided schema. If an object"
-        " it is omitted. The function also escapes any double quotes within JSON"\
+        " or field specified in the schema isn't contained within the correct JSON,"
-        " string values to ensure that they are valid. If the JSON string contains"\
+        " it is omitted. The function also escapes any double quotes within JSON"
+        " string values to ensure that they are valid. If the JSON string contains"
         " any None or NaN values, they are replaced with null before being parsed."
+    )
 
     # If it doesn't already start with a "`", add one:
     if not json_string.startswith("`"):

autogpt/llm_utils.py

@@ -126,13 +126,16 @@ def create_embedding_with_ada(text) -> list:
         backoff = 2 ** (attempt + 2)
         try:
             if CFG.use_azure:
-                return openai.Embedding.create(input=[text],
+                return openai.Embedding.create(
-                    engine=CFG.get_azure_deployment_id_for_model("text-embedding-ada-002"),
+                    input=[text],
+                    engine=CFG.get_azure_deployment_id_for_model(
+                        "text-embedding-ada-002"
+                    ),
                 )["data"][0]["embedding"]
             else:
-                return openai.Embedding.create(input=[text], model="text-embedding-ada-002")[
+                return openai.Embedding.create(
-                    "data"
+                    input=[text], model="text-embedding-ada-002"
-                ][0]["embedding"]
+                )["data"][0]["embedding"]
         except RateLimitError:
             pass
         except APIError as e:
@@ -148,4 +151,3 @@ def create_embedding_with_ada(text) -> list:
                 f"API Bad gateway. Waiting {backoff} seconds..." + Fore.RESET,
             )
         time.sleep(backoff)
-

autogpt/permanent_memory/sqlite3_store.py

@@ -18,11 +18,13 @@ class MemoryDB:
                 # As last resort, open in dynamic memory. Won't be persistent.
                 self.db_file = ":memory:"
             self.cnx = sqlite3.connect(self.db_file)
-            self.cnx.execute("CREATE VIRTUAL TABLE \
+            self.cnx.execute(
+                "CREATE VIRTUAL TABLE \
                 IF NOT EXISTS text USING FTS5 \
                 (session, \
                  key, \
-                 block);")
+                 block);"
+            )
             self.session_id = int(self.get_max_session_id()) + 1
             self.cnx.commit()
 
@@ -66,7 +68,7 @@ class MemoryDB:
             cnx = self.get_cnx()
             cnx.execute(cmd_str, (session_id, key, text))
             cnx.commit()
-    
+
     # Overwrite text at key.
     def overwrite(self, key, text):
         self.delete_memory(key)
@@ -76,8 +78,8 @@ class MemoryDB:
         cnx = self.get_cnx()
         cnx.execute(cmd_str, (session_id, key, text))
         cnx.commit()
-    
+
-    def delete_memory(self, key, session_id = None):
+    def delete_memory(self, key, session_id=None):
         session = session_id
         if session is None:
             session = self.session_id