rename the file containing the default models

wandb · Feb 5, 2025 · 919d0f3 · 919d0f3
1 parent 7c1ed3a
commit 919d0f3
Show file tree

Hide file tree

Showing 8 changed files with 9 additions and 7 deletions.
diff --git a/tests/scorers/test_similarity_scorer.py b/tests/scorers/test_similarity_scorer.py
@@ -3,7 +3,7 @@
 import pytest
 
 import weave
-from weave.scorers.llm_utils import OPENAI_DEFAULT_EMBEDDING_MODEL
+from weave.scorers.default_models import OPENAI_DEFAULT_EMBEDDING_MODEL
 from weave.scorers.similarity_scorer import (
     EmbeddingSimilarityScorer,
 )

diff --git a/weave/scorers/llm_utils.py → weave/scorers/default_models.py b/weave/scorers/llm_utils.py → weave/scorers/default_models.py
@@ -1,3 +1,5 @@
+"""Default model configurations for different LLM providers."""
+
 OPENAI_DEFAULT_MODEL = "gpt-4o"
 OPENAI_DEFAULT_EMBEDDING_MODEL = "text-embedding-3-small"
 OPENAI_DEFAULT_MODERATION_MODEL = "omni-moderation-latest"

diff --git a/weave/scorers/guardrails/prompt_injection_guardrail.py b/weave/scorers/guardrails/prompt_injection_guardrail.py
@@ -4,12 +4,12 @@
 from pydantic import BaseModel
 
 import weave
+from weave.scorers.default_models import OPENAI_DEFAULT_MODEL
 from weave.scorers.guardrails.prompts import (
     PROMPT_INJECTION_GUARDRAIL_SYSTEM_PROMPT,
     PROMPT_INJECTION_SURVEY_PAPER_SUMMARY,
 )
 from weave.scorers.llm_scorer import LLMScorer
-from weave.scorers.llm_utils import OPENAI_DEFAULT_MODEL
 from weave.scorers.utils import stringify
 
 

diff --git a/weave/scorers/hallucination_scorer.py b/weave/scorers/hallucination_scorer.py
@@ -2,8 +2,8 @@
 from pydantic import BaseModel, Field
 
 import weave
+from weave.scorers.default_models import OPENAI_DEFAULT_MODEL
 from weave.scorers.llm_scorer import LLMScorer
-from weave.scorers.llm_utils import OPENAI_DEFAULT_MODEL
 from weave.scorers.utils import stringify
 
 DEFAULT_HALLUCINATION_SYSTEM_PROMPT = """

diff --git a/weave/scorers/moderation_scorer.py b/weave/scorers/moderation_scorer.py
@@ -3,7 +3,7 @@
 from litellm import amoderation
 
 import weave
-from weave.scorers.llm_utils import OPENAI_DEFAULT_MODERATION_MODEL
+from weave.scorers.default_models import OPENAI_DEFAULT_MODERATION_MODEL
 
 
 class OpenAIModerationScorer(weave.Scorer):

diff --git a/weave/scorers/ragas_scorer.py b/weave/scorers/ragas_scorer.py
@@ -6,8 +6,8 @@
 from pydantic import BaseModel, Field
 
 import weave
+from weave.scorers.default_models import OPENAI_DEFAULT_MODEL
 from weave.scorers.llm_scorer import LLMScorer
-from weave.scorers.llm_utils import OPENAI_DEFAULT_MODEL
 
 
 class EntityExtractionResponse(BaseModel):

diff --git a/weave/scorers/similarity_scorer.py b/weave/scorers/similarity_scorer.py
@@ -6,7 +6,7 @@
 from pydantic import Field
 
 import weave
-from weave.scorers.llm_utils import OPENAI_DEFAULT_EMBEDDING_MODEL
+from weave.scorers.default_models import OPENAI_DEFAULT_EMBEDDING_MODEL
 
 
 class EmbeddingSimilarityScorer(weave.Scorer):

diff --git a/weave/scorers/summarization_scorer.py b/weave/scorers/summarization_scorer.py
@@ -5,8 +5,8 @@
 from pydantic import BaseModel, Field
 
 import weave
+from weave.scorers.default_models import OPENAI_DEFAULT_MODEL
 from weave.scorers.llm_scorer import LLMScorer
-from weave.scorers.llm_utils import OPENAI_DEFAULT_MODEL
 
 DEFAULT_EXTRACTION_SYSTEM_PROMPT = """
 Given a <text>, extract all the unique entities from the text without repetition.