providers updated

Author: staru09

core/llm_provider.py

@@ -46,7 +46,8 @@ def chat_completion(self, messages: list, model: str, max_tokens: int = 512) ->
             messages=messages,
             max_tokens=max_tokens,
         )
-        return response.choices[0].message.content
+        # LiteLLM returns an OpenAI-compatible dict-like response
+        return response["choices"][0]["message"]["content"]
 
 def get_llm_provider(provider_type: str, api_key: str) -> LLMProvider:
     """Factory function to get the appropriate LLM provider"""

main.py

@@ -3,7 +3,7 @@
 
 def main():
     parser = argparse.ArgumentParser(
-        description="Generate output using OpenAI's API")
+        description="Generate output using OpenAI's API or custom LLM using LiteLLM")
     parser.add_argument('--json_file', type=str, required=True,
                         help="Path to the JSON file with input data")