Ruff

neo4j · Jan 6, 2025 · d4133ab · d4133ab
1 parent 4f4b397
commit d4133ab
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 15 deletions.
diff --git a/examples/customize/llms/llm_with_message_history.py b/examples/customize/llms/llm_with_message_history.py
@@ -4,6 +4,7 @@
 
 OpenAILLM can be replaced by any supported LLM from this package.
 """
+
 from neo4j_graphrag.llm import LLMResponse, OpenAILLM
 
 # set api key here on in the OPENAI_API_KEY env var
@@ -24,15 +25,19 @@
         question,
         message_history=history,
     )
-    history.append({
-        "role": "user",
-        "content": question,
-    })
-    history.append({
-        "role": "assistant",
-        "content": res.content,
-    })
+    history.append(
+        {
+            "role": "user",
+            "content": question,
+        }
+    )
+    history.append(
+        {
+            "role": "assistant",
+            "content": res.content,
+        }
+    )
 
-    print("#"*50, question)
+    print("#" * 50, question)
     print(res.content)
-    print("#"*50)
+    print("#" * 50)
diff --git a/examples/customize/llms/llm_with_system_instructions.py b/examples/customize/llms/llm_with_system_instructions.py
@@ -2,6 +2,7 @@
 
 OpenAILLM can be replaced by any supported LLM from this package.
 """
+
 from neo4j_graphrag.llm import LLMResponse, OpenAILLM
 
 # set api key here on in the OPENAI_API_KEY env var
@@ -11,7 +12,7 @@
 llm = OpenAILLM(
     model_name="gpt-4o",
     api_key=api_key,
-    system_instruction="Answer with a serious tone"
+    system_instruction="Answer with a serious tone",
 )
 
 question = "How fast is Santa Claus during the Christmas eve?"
@@ -23,8 +24,5 @@
 
 
 # system instruction can be overwritten for each invoke:
-res: LLMResponse = llm.invoke(
-    question,
-    system_instruction="Answer with a joke"
-)
+res: LLMResponse = llm.invoke(question, system_instruction="Answer with a joke")
 print(res.content)