import os
from dotenv import load_dotenv
from langchain_groq import ChatGroq
from langchain.schema import HumanMessage

# Load environment variables from .env
load_dotenv()

# Initialize the Groq chat model with Qwen
llm = ChatGroq(
    temperature=0,
    model_name="qwen-qwq-32b",  # Updated to working model
    groq_api_key=os.getenv("GROQ_API_KEY")
)

# Send a test message
response = llm([HumanMessage(content="Hello Groq, how fast are you?")])

print(response.content)