ruff formatting changes

Author: jgbradley1

docs/examples_notebooks/index_migration_to_v1.ipynb

@@ -202,10 +202,11 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "from graphrag.index.flows.generate_text_embeddings import generate_text_embeddings\n",
+    "\n",
     "from graphrag.cache.factory import CacheFactory\n",
     "from graphrag.callbacks.noop_workflow_callbacks import NoopWorkflowCallbacks\n",
     "from graphrag.config.embeddings import get_embedded_fields, get_embedding_settings\n",
-    "from graphrag.index.flows.generate_text_embeddings import generate_text_embeddings\n",
     "\n",
     "# We only need to re-run the embeddings workflow, to ensure that embeddings for all required search fields are in place\n",
     "# We'll construct the context and run this function flow directly to avoid everything else\n",

examples_notebooks/community_contrib/yfiles-jupyter-graphs/graph-visualization.ipynb

@@ -29,6 +29,9 @@
     "\n",
     "import pandas as pd\n",
     "import tiktoken\n",
+    "from graphrag.query.llm.oai.chat_openai import ChatOpenAI\n",
+    "from graphrag.query.llm.oai.embedding import OpenAIEmbedding\n",
+    "from graphrag.query.llm.oai.typing import OpenaiApiType\n",
     "\n",
     "from graphrag.query.context_builder.entity_extraction import EntityVectorStoreKey\n",
     "from graphrag.query.indexer_adapters import (\n",
@@ -38,9 +41,6 @@
     "    read_indexer_reports,\n",
     "    read_indexer_text_units,\n",
     ")\n",
-    "from graphrag.query.llm.oai.chat_openai import ChatOpenAI\n",
-    "from graphrag.query.llm.oai.embedding import OpenAIEmbedding\n",
-    "from graphrag.query.llm.oai.typing import OpenaiApiType\n",
     "from graphrag.query.structured_search.local_search.mixed_context import (\n",
     "    LocalSearchMixedContext,\n",
     ")\n",

tests/integration/storage/test_cosmosdb_storage.py

@@ -24,7 +24,7 @@
 async def test_find():
     storage = CosmosDBPipelineStorage(
         connection_string=WELL_KNOWN_COSMOS_CONNECTION_STRING,
-        database_name="testfind",
+        base_dir="testfind",
         container_name="testfindcontainer",
     )
     try:
@@ -70,7 +70,7 @@ async def test_find():
 async def test_child():
     storage = CosmosDBPipelineStorage(
         connection_string=WELL_KNOWN_COSMOS_CONNECTION_STRING,
-        database_name="testchild",
+        base_dir="testchild",
         container_name="testchildcontainer",
     )
     try:
@@ -83,7 +83,7 @@ async def test_child():
 async def test_clear():
     storage = CosmosDBPipelineStorage(
         connection_string=WELL_KNOWN_COSMOS_CONNECTION_STRING,
-        database_name="testclear",
+        base_dir="testclear",
         container_name="testclearcontainer",
     )
     try:
@@ -113,7 +113,7 @@ async def test_clear():
 async def test_get_creation_date():
     storage = CosmosDBPipelineStorage(
         connection_string=WELL_KNOWN_COSMOS_CONNECTION_STRING,
-        database_name="testclear",
+        base_dir="testclear",
         container_name="testclearcontainer",
     )
     try: