Change prompt constants to UPPERCASE

langchain-ai · Feb 10, 2025 · 3bbf258 · 3bbf258
1 parent a0b5a05
commit 3bbf258
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/libs/langchain-mongodb/langchain_mongodb/graphrag/graph.py b/libs/langchain-mongodb/langchain_mongodb/graphrag/graph.py
@@ -113,7 +113,7 @@ def __init__(
             allowed_entity_types: If provided, constrains search to these types.
             allowed_relationship_types: If provided, constrains search to these types.
             entity_examples: A string containing any number of additional examples to provide as context for entity extraction.
-            entity_name_examples: A string appended to schema.name_extraction_instructions containing examples.
+            entity_name_examples: A string appended to prompts.NAME_EXTRACTION_INSTRUCTIONS containing examples.
             validate: If True, entity schema will be validated on every insert or update.
             validation_action: One of {"warn", "error"}.
               - If "warn", the default, documents will be inserted but errors logged.
@@ -204,7 +204,7 @@ def from_connection_string(
             allowed_entity_types: If provided, constrains search to these types.
             allowed_relationship_types: If provided, constrains search to these types.
             entity_examples: A string containing any number of additional examples to provide as context for entity extraction.
-            entity_name_examples: A string appended to schema.name_extraction_instructions containing examples.
+            entity_name_examples: A string appended to prompts.NAME_EXTRACTION_INSTRUCTIONS containing examples.
             validate: If True, entity schema will be validated on every insert or update.
             validation_action: One of {"warn", "error"}.
               - If "warn", the default, documents will be inserted but errors logged.

diff --git a/libs/langchain-mongodb/langchain_mongodb/graphrag/prompts.py b/libs/langchain-mongodb/langchain_mongodb/graphrag/prompts.py
@@ -4,7 +4,7 @@
     SystemMessagePromptTemplate,
 )
 
-entity_extraction_instructions = """
+ENTITY_EXTRACTION_INSTRUCTIONS = """
 ## Overview
 You are a meticulous analyst tasked with extracting information from unstructured text
 to build a knowledge graph in a structured json format of entities (nodes) and their relationships (edges).
@@ -81,7 +81,7 @@
 """
 
 
-name_extraction_instructions = """
+NAME_EXTRACTION_INSTRUCTIONS = """
 You are a meticulous analyst tasked with extracting information from documents to form
 knowledge graphs of entities (nodes) and their relationships (edges).
 
@@ -129,7 +129,7 @@
 """
 
 
-rag_instructions = """
+RAG_INSTRUCTIONS = """
 ## Context
 You are a meticulous analyst tasked with extracting information in the form of knowledge graphs
 comprised of entities (nodes) and their relationships (edges).
@@ -151,21 +151,21 @@
 
 entity_prompt = ChatPromptTemplate.from_messages(
     [
-        SystemMessagePromptTemplate.from_template(entity_extraction_instructions),
+        SystemMessagePromptTemplate.from_template(ENTITY_EXTRACTION_INSTRUCTIONS),
         HumanMessagePromptTemplate.from_template("{input_document}"),
     ]
 )
 
 query_prompt = ChatPromptTemplate.from_messages(
     [
-        SystemMessagePromptTemplate.from_template(name_extraction_instructions),
+        SystemMessagePromptTemplate.from_template(NAME_EXTRACTION_INSTRUCTIONS),
         HumanMessagePromptTemplate.from_template("{input_document}"),
     ]
 )
 
 rag_prompt = ChatPromptTemplate.from_messages(
     [
-        SystemMessagePromptTemplate.from_template(rag_instructions),
+        SystemMessagePromptTemplate.from_template(RAG_INSTRUCTIONS),
         HumanMessagePromptTemplate.from_template("{query}"),
     ]
 )