From 50398ad1f65692904949b9a71bf41225fc1e0122 Mon Sep 17 00:00:00 2001
From: emmaver <emma.verkinderen@hotmail.com>
Date: Wed, 13 Dec 2023 09:12:22 +0100
Subject: [PATCH] small fixes in prompts.py

---
 biochatter/prompts.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/biochatter/prompts.py b/biochatter/prompts.py
index b7bd1132..a544013e 100644
--- a/biochatter/prompts.py
+++ b/biochatter/prompts.py
@@ -193,7 +193,7 @@ def _select_entities(self, question: str) -> bool:
         self.question = question
 
         conversation = GptConversation(
-            model_name="gpt-3.5-turbo",
+            model_name=self.model_name,
             prompts={},
             correct=False,
         )
@@ -498,7 +498,7 @@ def _generate_query(
             for key, value in self.rel_directions.items():
                 for pair in value:
                     msg += f"'(:{pair[0]})-(:{key})->(:{pair[1]})', "
-            msg += "generate a Cypher query using one of these combinations. "
+            msg += f"generate a {query_language} query using one of these combinations. "
 
         msg += "Only return the query, without any additional text."