Upload llm.py

llm.py

@@ -22,14 +22,15 @@ def invoke_llm(context, task, model):
                            temperature=0.3,
                            openai_api_key=os.environ['OPENAI_KEY'],
                            max_tokens=1000)
+        result = llm.invoke(prompt)
+        return result.content
     else:
-        llm=ChatOpenAI(model_name="mistralai/Mixtral-8x7B-Instruct-v0.1",
+        llm=Together(model="mistralai/Mixtral-8x7B-Instruct-v0.1",
                            temperature=0.3,
                            together_api_key=os.environ['TOGETHER_KEY'],
                            max_tokens=1000)
-
-    result = llm.invoke(prompt)
-    return result.content
+        result = llm.invoke(prompt)
+        return result