Merge branch 'main' of https://github.com/biocypher/biochatter

biocypher · Dec 13, 2023 · fa9cf75 · fa9cf75
2 parents f79ce6e + caf7af2
commit fa9cf75
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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."