
    'jP                         d Z ddlZddlmZ ddlmZ ddlmZmZ ddl	m
Z
 ddlmZ  ee          j        dz  Z eee
ej        d	          e            d
d          Z eeddedd           dS )uH  Tatasteel Postgres data agent — pydantic-ai agent definition.

Connects to the Tatasteel construction project database using credentials
from environment variables. The agent reads a schema definition to understand
the DB structure, generates accurate SQL queries, and executes them safely
(SELECT-only) to answer user questions.

Required env vars:
    TATASTEEL_DB_HOST     — database host (default: localhost)
    TATASTEEL_DB_PORT     — database port (default: 5432)
    TATASTEEL_DB_NAME     — database name
    TATASTEEL_DB_USER     — database user
    TATASTEEL_DB_PASSWORD — database password
    TATASTEEL_SCHEMA_PATH — path to the schema file
                            (default: backend/chanakya/tatasteel/schema.sql)

The ASGI app, HTTP handler, and lifecycle hooks live in
``backend.chanakya.tatasteel.server``.
    N)Path)EnterpriseAgent)
NL2SQLDepsregister_nl2sql_tools)TatasteelAgentOutput)make_tatasteel_instructionsz
schema.sqlLITELLM_PROVIDER_BASE_URLzchanakya-alpha	tatasteel)	deps_typeoutput_typeapi_baseinstructionslogfire_envservice_nameTATASTEEL_SCHEMA_PATHTATASTEEL_DBF)schema_path_envdb_env_prefixdefault_schema_pathregister_get_schemaregister_format_response)__doc__ospathlibr   backend.chanakya.baser   backend.chanakya.nl2sql_corer   r   backend.chanakya.schemasr   !backend.chanakya.tatasteel.promptr   __file__parent_DEFAULT_SCHEMA_PATHenvirontatasteel_agent     E/var/www/html/ai-enterprise-brain/backend/chanakya/tatasteel/agent.py<module>r'      s   & 
			       1 1 1 1 1 1 J J J J J J J J 9 9 9 9 9 9 I I I I I ItH~~,|; 
 "/$Z34,,..     + ,"     r%   