diff --git a/examples/conversational-rag/graph_db_example/application.py b/examples/conversational-rag/graph_db_example/application.py index fd2261cc..aa1c0187 100644 --- a/examples/conversational-rag/graph_db_example/application.py +++ b/examples/conversational-rag/graph_db_example/application.py @@ -66,7 +66,7 @@ def schema_to_prompt(schema): return prompt -def set_inital_chat_history(schema_prompt: str) -> list[dict]: +def set_initial_chat_history(schema_prompt: str) -> list[dict]: SYSTEM_MESSAGE = "You are a Cypher expert with access to a directed knowledge graph\n" SYSTEM_MESSAGE += schema_prompt SYSTEM_MESSAGE += ( @@ -212,7 +212,7 @@ def build_application( # create a prompt from it schema_prompt = schema_to_prompt(schema) # set the initial chat history - base_messages = set_inital_chat_history(schema_prompt) + base_messages = set_initial_chat_history(schema_prompt) tracker = LocalTrackingClient("ufc-falkor") # create graph diff --git a/examples/conversational-rag/graph_db_example/notebook.ipynb b/examples/conversational-rag/graph_db_example/notebook.ipynb index a246c3d0..45d8ba71 100644 --- a/examples/conversational-rag/graph_db_example/notebook.ipynb +++ b/examples/conversational-rag/graph_db_example/notebook.ipynb @@ -103,7 +103,7 @@ "\n", " return prompt\n", "\n", - "def set_inital_chat_history(schema_prompt: str) -> list[dict]:\n", + "def set_initial_chat_history(schema_prompt: str) -> list[dict]:\n", " \"\"\"Helper to set initial system message\"\"\"\n", " SYSTEM_MESSAGE = \"You are a Cypher expert with access to a directed knowledge graph\\n\"\n", " SYSTEM_MESSAGE += schema_prompt\n", @@ -363,7 +363,7 @@ "# create a prompt from it\n", "schema_prompt = schema_to_prompt(schema)\n", "# set the initial chat history\n", - "base_messages = set_inital_chat_history(schema_prompt)" + "base_messages = set_initial_chat_history(schema_prompt)" ], "outputs": [] },