From 37413df78cfd54122396a7a38d73c0ce2c923da4 Mon Sep 17 00:00:00 2001 From: Putu Gede Pradana Adnyana Date: Mon, 7 Jul 2025 11:26:04 +0800 Subject: [PATCH 1/2] Update prompt.py --- .../langchain_mongodb/agent_toolkit/prompt.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/libs/langchain-mongodb/langchain_mongodb/agent_toolkit/prompt.py b/libs/langchain-mongodb/langchain_mongodb/agent_toolkit/prompt.py index 37e40afa..f722135a 100644 --- a/libs/langchain-mongodb/langchain_mongodb/agent_toolkit/prompt.py +++ b/libs/langchain-mongodb/langchain_mongodb/agent_toolkit/prompt.py @@ -34,17 +34,25 @@ MONGODB_FUNCTIONS_SUFFIX = """I should look at the collections in the database to see what I can query. Then I should query the schema of the most relevant collections.""" -MONGODB_QUERY_CHECKER = """ +MONGODB_QUERY_CHECKER = MONGODB_QUERY_CHECKER = """ {query} Double check the MongoDB query above for common mistakes, including: - Missing content in the aggegregation pipeline - Improperly quoting identifiers - Improperly quoting operators -- The content in the aggregation pipeline is not valid JSON - +The query must be using pymongo syntax, so make sure the syntax is correct. +Example: +```python +db.CollectionName.aggregate([ + {{ "$match": {{ "field": "value" }} }}, + {{ "$group": {{ "_id": "$field", "count": {{ "$sum": 1 }} }} }}, + {{ "$sort": {{ "count": -1 }} }} +]) +``` If there are any of the above mistakes, rewrite the query. If there are no mistakes, just reproduce the original query. Output the final MongoDB query only. MongoDB Query: """ + From 27a59c367f73a4732428b1eca5efeca6051dd5de Mon Sep 17 00:00:00 2001 From: Putu Gede Pradana Adnyana Date: Mon, 7 Jul 2025 11:32:43 +0800 Subject: [PATCH 2/2] Update prompt.py --- .../langchain-mongodb/langchain_mongodb/agent_toolkit/prompt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/langchain-mongodb/langchain_mongodb/agent_toolkit/prompt.py b/libs/langchain-mongodb/langchain_mongodb/agent_toolkit/prompt.py index f722135a..f218cd13 100644 --- a/libs/langchain-mongodb/langchain_mongodb/agent_toolkit/prompt.py +++ b/libs/langchain-mongodb/langchain_mongodb/agent_toolkit/prompt.py @@ -41,7 +41,7 @@ - Missing content in the aggegregation pipeline - Improperly quoting identifiers - Improperly quoting operators -The query must be using pymongo syntax, so make sure the syntax is correct. +- The query must be using pymongo syntax, so make sure the syntax is correct. Example: ```python db.CollectionName.aggregate([