diff --git a/oracles/src/entities.py b/oracles/src/entities.py
index 6bc648c..6132f81 100644
--- 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]