Ruff 🐶

Arize-ai · Dec 2, 2024 · a216250 · a216250
1 parent 8a365ce
commit a216250
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 15 deletions.
diff --git a/packages/phoenix-evals/src/phoenix/evals/models/anthropic.py b/packages/phoenix-evals/src/phoenix/evals/models/anthropic.py
@@ -4,7 +4,7 @@
 from phoenix.evals.exceptions import PhoenixContextLimitExceeded
 from phoenix.evals.models.base import BaseModel
 from phoenix.evals.models.rate_limiters import RateLimiter
-from phoenix.evals.templates import PromptPartContentType, MultimodalPrompt
+from phoenix.evals.templates import MultimodalPrompt, PromptPartContentType
 
 MINIMUM_ANTHROPIC_VERSION = "0.18.0"
 

diff --git a/packages/phoenix-evals/src/phoenix/evals/models/litellm.py b/packages/phoenix-evals/src/phoenix/evals/models/litellm.py
@@ -4,7 +4,7 @@
 from typing import Any, Dict, List, Optional, Union
 
 from phoenix.evals.models.base import BaseModel
-from phoenix.evals.templates import PromptPartContentType, MultimodalPrompt
+from phoenix.evals.templates import MultimodalPrompt, PromptPartContentType
 
 logger = logging.getLogger(__name__)
 

diff --git a/packages/phoenix-evals/src/phoenix/evals/models/mistralai.py b/packages/phoenix-evals/src/phoenix/evals/models/mistralai.py
@@ -3,7 +3,7 @@
 
 from phoenix.evals.models.base import BaseModel
 from phoenix.evals.models.rate_limiters import RateLimiter
-from phoenix.evals.templates import PromptPartContentType, MultimodalPrompt
+from phoenix.evals.templates import MultimodalPrompt, PromptPartContentType
 
 if TYPE_CHECKING:
     from mistralai.models.chat_completion import ChatMessage

diff --git a/packages/phoenix-evals/src/phoenix/evals/models/openai.py b/packages/phoenix-evals/src/phoenix/evals/models/openai.py
@@ -17,7 +17,7 @@
 from phoenix.evals.exceptions import PhoenixContextLimitExceeded
 from phoenix.evals.models.base import BaseModel
 from phoenix.evals.models.rate_limiters import RateLimiter
-from phoenix.evals.templates import PromptPartContentType, MultimodalPrompt
+from phoenix.evals.templates import MultimodalPrompt, PromptPartContentType
 
 MINIMUM_OPENAI_VERSION = "1.0.0"
 MODEL_TOKEN_LIMIT_MAPPING = {

diff --git a/packages/phoenix-evals/src/phoenix/evals/templates.py b/packages/phoenix-evals/src/phoenix/evals/templates.py
@@ -53,18 +53,12 @@ class MultimodalPrompt:
     @staticmethod
     def from_string(string_prompt: str) -> "MultimodalPrompt":
         return MultimodalPrompt(
-            parts=[
-                PromptPart(content_type=PromptPartContentType.TEXT, content=string_prompt)
-            ]
+            parts=[PromptPart(content_type=PromptPartContentType.TEXT, content=string_prompt)]
         )
 
     def to_text_only_prompt(self) -> str:
         return "\n\n".join(
-            [
-                part.content
-                for part in self.parts
-                if part.content_type == PromptPartContentType.TEXT
-            ]
+            [part.content for part in self.parts if part.content_type == PromptPartContentType.TEXT]
         )
 
     def __str__(self) -> str:
@@ -122,9 +116,7 @@ def _normalize_template(
         self, template: Union[str, List[PromptPartTemplate]]
     ) -> List[PromptPartTemplate]:
         if isinstance(template, str):
-            return [
-                PromptPartTemplate(content_type=PromptPartContentType.TEXT, template=template)
-            ]
+            return [PromptPartTemplate(content_type=PromptPartContentType.TEXT, template=template)]
         return template