chat_service.py 357 B

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