together and mistral language model wrappers to no longer delete afte…

…r last ".". PiperOrigin-RevId: 685414218 Change-Id: I9afa0e49c1610abbe920dd2a5376e7bdc2e08744
google-deepmind · Oct 13, 2024 · afbecd7 · afbecd7
1 parent 89ed637
commit afbecd7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/concordia/language_model/mistral_model.py b/concordia/language_model/mistral_model.py
@@ -126,10 +126,7 @@ def _complete_text(
           self._channel,
           {'raw_text_length': len(result)},
       )
-    # Remove the occasional sentence fragment from the end of the result.
-    last_stop = result.rfind('.')
-    if last_stop >= 0:
-      result = result[:last_stop + 1]
+
     return result
 
   def _chat_text(

diff --git a/concordia/language_model/together_ai.py b/concordia/language_model/together_ai.py
@@ -204,10 +204,7 @@ def sample_text(
           self._channel,
           {'raw_text_length': len(result)},
       )
-    # Remove the occasional sentence fragment from the end of the result.
-    last_stop = result.rfind('.')
-    if last_stop >= 0:
-      result = result[: last_stop + 1]
+
     return result
 
   @override