diff --git a/standalone_llm_tool.py b/standalone_llm_tool.py index 3dac90e..834b913 100644 --- a/standalone_llm_tool.py +++ b/standalone_llm_tool.py @@ -23,11 +23,9 @@ class StandaloneLLMTool: logging.info("Response generated") return response.choices[0].text -def main(): - tool = StandaloneLLMTool() - user_prompt = input("Enter your prompt: ") - response = tool.process_user_input(user_prompt) - print("Response:", response) -if __name__ == '__main__': - main() +# Utility function for programmatic access + +def get_llm_response(prompt): + tool = StandaloneLLMTool() + return tool.process_user_input(prompt)