Skip to content

Add Groq support#10

Open
bitsnaps wants to merge 2 commits into
Doriandarko:mainfrom
bitsnaps:dev
Open

Add Groq support#10
bitsnaps wants to merge 2 commits into
Doriandarko:mainfrom
bitsnaps:dev

Conversation

@bitsnaps
Copy link
Copy Markdown

The most important part is:

# DEFAULT_MAX_TOKENS=8000

client = None
if os.getenv("OPENROUTER_API_KEY"):
    client = OpenAI(
        base_url="https://openrouter.ai/api/v1",
        api_key=os.getenv("OPENROUTER_API_KEY")
    )
    # DEFAULT_MAX_TOKENS=2000
elif os.getenv("GROQ_API_KEY"):
    client = OpenAI(
        base_url="https://api.groq.com/openai/v1",
        api_key=os.getenv("GROQ_API_KEY")
    )
    DEFAULT_MODEL = "llama-3.1-70b-versatile"
    EDITOR_MODEL = "llama-3.1-70b-versatile"
else:
    client = OpenAI(
        api_key=os.getenv("OPENAI_API_KEY")
    )

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant