File size: 620 Bytes
7236cee 6919e51 7236cee 6919e51 7236cee 6919e51 7236cee bc06af6 7236cee c7eea5d 6919e51 7236cee | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | from langchain.chat_models import init_chat_model
from dotenv import load_dotenv
from openai import AsyncOpenAI
import os
load_dotenv()
groq_client = AsyncOpenAI(
base_url="https://api.groq.com/openai/v1",
api_key=os.getenv("GROQ_API_KEY"),
)
llm_supervisor = init_chat_model(
model="groq:openai/gpt-oss-120b",
max_tokens=1000
)
llm_peripheral = init_chat_model(
model="groq:llama-3.1-8b-instant"
)
llm_sub_agents = init_chat_model(
model="groq:moonshotai/kimi-k2-instruct-0905",
max_tokens=4000
)
llm_image = init_chat_model(
model="groq:meta-llama/llama-4-scout-17b-16e-instruct"
) |