Update app.py

Revised Helper Function
Direct Model Invocation:
The LiteLLMModel instance is likely callable. Instead of using a .complete() method, call the model directly with the prompt:
Response Handling:
The returned result might be a string or a dictionary. Check the type of result and extract the text accordingly. If your specific model returns responses in a different format, adjust the helper function to handle that format.
Consistency:
Update all parts of your code where you previously used .complete() to use the new call_llm function or a direct call to llm.

app.py

@@ -48,10 +48,15 @@ def call_llm(prompt: str) -> str:
     """
     Helper to call the LLM with a prompt, handling response extraction.
     """
-    result = llm.complete(prompt)
-    if isinstance(result, dict):
-        return result.get('text', '')
-    return result
+    try:
+        result = llm(prompt)  # Directly call the model with the prompt
+        # If the result is a dictionary and contains 'text', extract it.
+        if isinstance(result, dict) and 'text' in result:
+            return result['text']
+        return result if isinstance(result, str) else str(result)
+    except Exception as e:
+        logger.error(f"LLM call error: {e}")
+        return ""
 
 ###############################################################################
 # 3) CSV Loading and Processing