from groq import Groq

# your API key
client = Groq(api_key="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")

MODEL = "openai/gpt-oss-20b"   # change here if needed



def get1(prompt):
    response = client.chat.completions.create(
        model=MODEL,
        messages=[
            {"role": "user", "content": prompt}
        ]
    )

    return response.choices[0].message.content



if __name__ == "__main__":
    user_prompt = input("You: ")
    answer = get1(user_prompt)
    print("\nLLM:", answer, "\n")