Fix path

Kiln-AI · Jan 31, 2025 · 6e4c693 · 6e4c693
1 parent f38450e
commit 6e4c693
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 4 deletions.
diff --git a/libs/core/kiln_ai/adapters/model_adapters/openai_model_adapter.py b/libs/core/kiln_ai/adapters/model_adapters/openai_model_adapter.py
@@ -1,5 +1,9 @@
 from typing import Any, Dict, NoReturn
 
+from openai import AsyncOpenAI
+from openai.types.chat import ChatCompletion, ChatCompletionMessage
+from openai.types.chat.chat_completion import Choice
+
 import kiln_ai.datamodel as datamodel
 from kiln_ai.adapters.ml_model_list import StructuredOutputMode
 from kiln_ai.adapters.model_adapters.base_adapter import (
@@ -9,9 +13,6 @@
     RunOutput,
 )
 from kiln_ai.adapters.parsers.json_parser import parse_json_string
-from openai import AsyncOpenAI
-from openai.types.chat import ChatCompletion, ChatCompletionMessage
-from openai.types.chat.chat_completion import Choice
 
 
 class OpenAICompatibleAdapter(BaseAdapter):

diff --git a/libs/core/kiln_ai/adapters/parsers/test_base_parser.py b/libs/core/kiln_ai/adapters/parsers/test_base_parser.py
@@ -1,4 +1,5 @@
 import pytest
+
 from kiln_ai.adapters.parsers.json_parser import parse_json_string
 
 

diff --git a/libs/core/kiln_ai/adapters/parsers/test_r1_parser.py b/libs/core/kiln_ai/adapters/parsers/test_r1_parser.py
@@ -1,4 +1,5 @@
 import pytest
+
 from kiln_ai.adapters.parsers.r1_parser import R1ThinkingParser
 from kiln_ai.adapters.run_output import RunOutput
 

diff --git a/libs/core/kiln_ai/adapters/test_prompt_builders.py b/libs/core/kiln_ai/adapters/test_prompt_builders.py
@@ -3,6 +3,9 @@
 import pytest
 
 from kiln_ai.adapters.model_adapters.base_adapter import AdapterInfo, BaseAdapter
+from kiln_ai.adapters.model_adapters.test_structured_output import (
+    build_structured_output_test_task,
+)
 from kiln_ai.adapters.prompt_builders import (
     FewShotChainOfThoughtPromptBuilder,
     FewShotPromptBuilder,
@@ -16,7 +19,6 @@
     prompt_builder_from_ui_name,
 )
 from kiln_ai.adapters.test_prompt_adaptors import build_test_task
-from kiln_ai.adapters.test_structured_output import build_structured_output_test_task
 from kiln_ai.datamodel import (
     DataSource,
     DataSourceType,
Original file line number	Diff line number	Diff line change
		@@ -1,4 +1,5 @@
		import pytest

		from kiln_ai.adapters.parsers.json_parser import parse_json_string


Expand Down