Skip to content

Commit

Permalink
chore: update groq models
Browse files Browse the repository at this point in the history
  • Loading branch information
CNSeniorious000 committed Dec 8, 2024
1 parent 2d63760 commit ad07c50
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 2 deletions.
6 changes: 4 additions & 2 deletions src/utils/llm/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,10 +69,12 @@
"llama-3.1-70b-versatile",
"llama-3.1-70b-specdec",
"llama-3.1-405b-reasoning",
"llama-3.2-1b-text-preview",
"llama-3.2-3b-text-preview",
"llama-3.2-1b-preview",
"llama-3.2-3b-preview",
"llama-3.2-11b-vision-preview",
"llama-3.2-90b-vision-preview",
"llama-3.3-70b-versatile",
"llama-3.3-70b-specdec",
"llama3.1-8b",
"llama3.1-70b",
"mixtral-8x7b-32768",
Expand Down
1 change: 1 addition & 0 deletions src/utils/llm/groq.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@
@link_llm("llama3-")
@link_llm("llama-3.1")
@link_llm("llama-3.2")
@link_llm("llama-3.3")
@link_llm("mixtral")
class Groq(AsyncChatOpenAI):
async def complete(self, prompt: str | list[Message], /, **config):
Expand Down

0 comments on commit ad07c50

Please sign in to comment.