j dZddlmZddlZddlZddlmZddlmZm Z m Z ddl m Z ejeZdZd dZddd  d d Zy) z author: Yagnik Poshiya organization: Webbrains Technologies Private Limited Dedicated Mongo query generation LLM for live MongoDB connections. ) annotationsN)Settings)GenerationConfig LLMClientLLMUsage)build_llm_clientakYou are a specialized MongoDB query generator. Your ONLY job is to translate a visitor's natural-language question into a single safe MongoDB query spec JSON object. OUTPUT FORMAT (strict JSON object only): { "operation": "find" | "aggregate", "collection": "", "filter": { ... }, // for find "projection": { ... }, // optional for find "sort": [["field","asc"|"desc"]], // optional for find "limit": 1..50, // optional for find (default 50) "pipeline": [ ... ] // for aggregate } RULES: 1. Output MUST be valid JSON with exactly one query spec object. 2. Use ONLY collections and fields provided in schema context. 3. Never use write/update/delete/admin operations. 4. Allowed operations are only "find" and "aggregate". 5. For count-like questions, prefer aggregate with $match + $count. 6. Add a limit <= 50 when returning documents. 7. Do not use JavaScript execution operators like $where. 8. Return {"operation":"", "reason":"..."} only when schema lacks required data. 9. Do NOT output markdown fences or extra commentary. cBt|j|jS)N)rsql_gen_provider sql_gen_model)settingss M/var/www/html/agent-engine/apps/agents/rag-agent/src/agent/mongo_generator.py_build_mongo_gen_clientr0s H55x7M7M NN) previous_specprevious_errorc t|}|r|rd|d|d|d|d|d }n d|d|d|}td|}|j|tdd  } | jj } | sid | j fS tj| } t| tsi| | j fS| | | j fS#tj$r+tjd | ddi| | j fcYSwxYw)z/Return (query_spec_dict, raw_json_text, usage).z Database: z Schema: z Visitor's question: z Previous query spec: z Problem with previous spec: z( Generate a corrected Mongo query spec.z g333333?T) temperature response_json)configz&mongo_generator: non-JSON response: %sN)r_SYSTEM_INSTRUCTION_MONGOgeneratertextstripusagejsonloadsJSONDecodeErrorloggerwarning isinstancedict) r user_message schema_detaildb_labelrrclient user_blockpromptresprdatas r generate_mongo_queryr,5sI%X .F #%'$$0>2%%2O4,,:+;<5  5  #%'$$0> 3  **$zl ;F ??6*:t[_*`? aD 99?? D 2tzz!!$zz$ dD !4## tzz !!   $?dsL4##$sC;DD)r rreturnr)r rr$strr%r.r&r.r str | Nonerr/r-ztuple[dict, str, LLMUsage])__doc__ __future__rrloggingconfig.settingsrllm.baserrr llm.factoryr getLogger__name__r rrr,rr r9s# $::(   8 $:O!%!%("("(" ("  ("  ("(" ("r