lint

capeprivacy · Sep 26, 2023 · e4a36de · e4a36de
1 parent 1bb6b5b
commit e4a36de
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/examples/llama/llama.py b/examples/llama/llama.py
@@ -1,13 +1,18 @@
-from pycape.llms import Cape
 import os
 
+from pycape.llms import Cape
+
 c = Cape(url="https://api.capeprivacy.com")
 
 token = c.token(os.getenv("CAPE_TOKEN", ""))
 
 for msg in c.chat_completions(
     [
-        {"role": "user", "content": "<s>[INST] <<SYS>>You are a helpful Assistant.<</SYS>>\n\nWhat is the Capital of France? [/INST]"},
+        {
+            "role": "user",
+            "content": "<s>[INST] <<SYS>>You are a helpful Assistant.<</SYS>>"
+            "\n\nWhat is the Capital of France? [/INST]",
+        },
         {"role": "system", "content": "you are a happy helpful assistant"},
     ],
     token,