from agent.db_agent import invoke class ChatService: @staticmethod def generate_sql(requirement: str, schema: str, source_type: str = "hive") -> object: try: result = invoke(requirement, schema, source_type) return result except Exception as e: raise RuntimeError(f"LLM service error: {str(e)}")