Small type fixes (#53)

galadriel-ai · Apr 2, 2024 · b517a2f · b517a2f
1 parent 154f770
commit b517a2f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/oracles/src/entities.py b/oracles/src/entities.py
@@ -20,7 +20,7 @@ class PromptType(str, Enum):
     GROQ = "Groq"
 
 
-PromptTypeLiteral = Literal[PromptType.DEFAULT, PromptType.OPENAI]
+PromptTypeLiteral = Literal[PromptType.DEFAULT, PromptType.OPENAI, PromptType.GROQ]
 
 OpenaiToolChoiceType = Literal["none", "auto"]
 
@@ -47,7 +47,7 @@ class OpenAiConfig:
 
 @dataclass
 class GroqConfig:
-    model: OpenAiModelType
+    model: GroqModelType
     frequency_penalty: Optional[float]
     logit_bias: Optional[Dict]
     max_tokens: Optional[int]