22

utils/prompt_utils.py

@@ -14,11 +14,10 @@ def polish_prompt(original_prompt):
 
     # client = InferenceClient(provider="cerebras", api_key=api_key)
     # messages = []
-    client = InferenceClient(provider="cerebras")
+    client = InferenceClient(provider="cerebras", api_key=api_key)
     
     try:
         completion = client.chat.completions.create(
-        api_key=api_key,
         model="Qwen/Qwen3-235B-A22B-Instruct-2507",
         messages=[
                 {"role": "system", "content": system_prompt},