Update GPT4KG.py

GPT4KG.py

@@ -111,7 +111,7 @@ class KnowledgeGraph:
       except:
         pass
       messages.append({"role":"user","content":text})
-      output = openai.ChatCompletion.create(model="gpt-4-1106-preview",messages=messages)["choices"][0]["message"].to_dict()["content"]
+      output = openai.ChatCompletion.create(model="gpt-4o",messages=messages)["choices"][0]["message"].to_dict()["content"]
       return output
 
     def learn(self,text,show_output=False):
@@ -204,7 +204,7 @@ class KnowledgeGraph:
         system = {"role":"system","content":"You are a helpful chatbot that only outputs YES or NO"}
         messages = [system]
         messages.append({"role":"user","content":f"Do these two facts in our database express the same thing?: {pair}"})
-        output = openai.ChatCompletion.create(model="gpt-4-1106-preview",messages=messages)["choices"][0]["message"].to_dict()["content"]
+        output = openai.ChatCompletion.create(model="gpt-4o",messages=messages)["choices"][0]["message"].to_dict()["content"]
         if "yes" in output.lower():
           bad_index = facts.index(pair[1])
           redundant = rels[bad_index]