Blacken, requirements

Author: martin-martin

prompt-engineering/app.py

@@ -5,6 +5,8 @@
 
 from openai import OpenAI
 
+__all__ = ["get_chat_completion"]
+
 # Authenticate
 client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
@@ -30,23 +32,35 @@ def get_chat_completion(content: str) -> str:
     """Send a request to the /chat/completions endpoint."""
     response = client.chat.completions.create(
         model=SETTINGS["general"]["model"],
-        messages=assemble_chat_messages(content),
+        messages=_assemble_chat_messages(content),
         temperature=SETTINGS["general"]["temperature"],
         seed=12345,  # Doesn't do anything for older models
     )
     return response.choices[0].message.content
 
 
-def assemble_chat_messages(content: str) -> list[dict]:
+def _assemble_chat_messages(content: str) -> list[dict]:
     """Combine all messages into a well-formatted list of dicts."""
     messages = [
         {"role": "system", "content": SETTINGS["prompts"]["role_prompt"]},
         {"role": "user", "content": SETTINGS["prompts"]["negative_example"]},
-        {"role": "system", "content": SETTINGS["prompts"]["negative_reasoning"]},
-        {"role": "assistant", "content": SETTINGS["prompts"]["negative_output"]},
+        {
+            "role": "system",
+            "content": SETTINGS["prompts"]["negative_reasoning"],
+        },
+        {
+            "role": "assistant",
+            "content": SETTINGS["prompts"]["negative_output"],
+        },
         {"role": "user", "content": SETTINGS["prompts"]["positive_example"]},
-        {"role": "system", "content": SETTINGS["prompts"]["positive_reasoning"]},
-        {"role": "assistant", "content": SETTINGS["prompts"]["positive_output"]},
+        {
+            "role": "system",
+            "content": SETTINGS["prompts"]["positive_reasoning"],
+        },
+        {
+            "role": "assistant",
+            "content": SETTINGS["prompts"]["positive_output"],
+        },
         {"role": "user", "content": f">>>>>\n{content}\n<<<<<"},
         {"role": "user", "content": SETTINGS["prompts"]["instruction_prompt"]},
     ]

prompt-engineering/requirements.txt

@@ -1,15 +1,14 @@
-aiohttp==3.8.4
-aiosignal==1.3.1
-async-timeout==4.0.2
-attrs==23.1.0
-certifi==2023.5.7
-charset-normalizer==3.1.0
-frozenlist==1.3.3
-idna==3.4
-multidict==6.0.4
-openai==0.27.8
-python-dotenv==1.0.0
-requests==2.31.0
-tqdm==4.65.0
-urllib3==2.0.3
-yarl==1.9.2
+annotated-types==0.6.0
+anyio==4.3.0
+certifi==2024.2.2
+distro==1.9.0
+h11==0.14.0
+httpcore==1.0.4
+httpx==0.27.0
+idna==3.6
+openai==1.13.3
+pydantic==2.6.3
+pydantic-core==2.16.3
+sniffio==1.3.1
+tqdm==4.66.2
+typing-extensions==4.10.0