Apply latest Black, refs #400

simonw · Jan 26, 2024 · 2c87a6a · 2c87a6a
1 parent d9d5c72
commit 2c87a6a
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 13 deletions.
diff --git a/llm/cli.py b/llm/cli.py
@@ -746,12 +746,16 @@ def logs_list(
             click.echo(
                 "# {}{}\n{}".format(
                     row["datetime_utc"].split(".")[0],
-                    "    conversation: {}".format(row["conversation_id"])
-                    if should_show_conversation
-                    else "",
-                    "\nModel: **{}**\n".format(row["model"])
-                    if should_show_conversation
-                    else "",
+                    (
+                        "    conversation: {}".format(row["conversation_id"])
+                        if should_show_conversation
+                        else ""
+                    ),
+                    (
+                        "\nModel: **{}**\n".format(row["model"])
+                        if should_show_conversation
+                        else ""
+                    ),
                 )
             )
             # In conversation log mode only show it for the first one

diff --git a/llm/embeddings.py b/llm/embeddings.py
@@ -220,12 +220,12 @@ def embed_multi_with_metadata(
                             "collection_id": collection_id,
                             "id": id,
                             "embedding": llm.encode(embedding),
-                            "content": value
-                            if (store and isinstance(value, str))
-                            else None,
-                            "content_blob": value
-                            if (store and isinstance(value, bytes))
-                            else None,
+                            "content": (
+                                value if (store and isinstance(value, str)) else None
+                            ),
+                            "content_blob": (
+                                value if (store and isinstance(value, bytes)) else None
+                            ),
                             "content_hash": self.content_hash(value),
                             "metadata": json.dumps(metadata) if metadata else None,
                             "updated": int(time.time()),

diff --git a/setup.py b/setup.py
@@ -55,7 +55,7 @@ def get_long_description():
             "pytest-httpx",
             "cogapp",
             "mypy",
-            "black",
+            "black>=24.1.0",
             "ruff",
             "types-click",
             "types-PyYAML",