- Added translate API

- bug fix during profile list

Author: aosingh

src/select_ai/action.py

@@ -21,3 +21,4 @@ class Action(StrEnum):
     SHOWPROMPT = "showprompt"
     FEEDBACK = "feedback"
     SUMMARIZE = "summarize"
+    TRANSLATE = "translate"

src/select_ai/async_profile.py

@@ -7,7 +7,6 @@
 
 import json
 from contextlib import asynccontextmanager
-from dataclasses import replace as dataclass_replace
 from typing import (
     AsyncGenerator,
     List,
@@ -32,7 +31,6 @@
 from select_ai.db import async_cursor, async_get_connection
 from select_ai.errors import (
     ProfileAttributesEmptyError,
-    ProfileExistsError,
     ProfileNotFoundError,
 )
 from select_ai.feedback import (
@@ -73,14 +71,14 @@ async def _init_profile(self):
         if self.profile_name:
             profile_exists = False
             try:
-                saved_attributes = await self._get_attributes(
-                    profile_name=self.profile_name,
-                    raise_on_empty=True,
-                )
                 saved_description = await self._get_profile_description(
                     profile_name=self.profile_name
                 )
                 profile_exists = True
+                saved_attributes = await self._get_attributes(
+                    profile_name=self.profile_name,
+                    raise_on_empty=True,
+                )
                 self._raise_error_if_profile_exists()
             except ProfileAttributesEmptyError:
                 if self.raise_error_on_empty_attributes:
@@ -624,6 +622,34 @@ async def run_pipeline(
                 responses.append(result.error)
         return responses
 
+    async def translate(
+        self, text: str, source_language: str, target_language: str
+    ) -> Union[str, None]:
+        """
+        Translate a text using a source language and a target language
+
+        :param str text: Text to translate
+        :param str source_language: Source language
+        :param str target_language: Target language
+        :return: str
+        """
+        parameters = {
+            "profile_name": self.profile_name,
+            "text": text,
+            "source_language": source_language,
+            "target_language": target_language,
+        }
+        async with async_cursor() as cr:
+            data = await cr.callfunc(
+                "DBMS_CLOUD_AI.TRANSLATE",
+                oracledb.DB_TYPE_CLOB,
+                keyword_parameters=parameters,
+            )
+        if data is not None:
+            result = await data.read()
+            return result
+        return None
+
 
 class AsyncSession:
     """AsyncSession lets you persist request parameters across DBMS_CLOUD_AI

src/select_ai/profile.py

@@ -7,7 +7,6 @@
 
 import json
 from contextlib import contextmanager
-from dataclasses import replace as dataclass_replace
 from typing import Generator, Mapping, Optional, Tuple, Union
 
 import oracledb
@@ -25,7 +24,6 @@
 from select_ai.db import cursor
 from select_ai.errors import (
     ProfileAttributesEmptyError,
-    ProfileExistsError,
     ProfileNotFoundError,
 )
 from select_ai.feedback import FeedbackOperation, FeedbackType
@@ -59,14 +57,14 @@ def _init_profile(self) -> None:
         if self.profile_name:
             profile_exists = False
             try:
-                saved_attributes = self._get_attributes(
-                    profile_name=self.profile_name,
-                    raise_on_empty=True,
-                )
                 saved_description = self._get_profile_description(
                     profile_name=self.profile_name
                 )
                 profile_exists = True
+                saved_attributes = self._get_attributes(
+                    profile_name=self.profile_name,
+                    raise_on_empty=True,
+                )
                 self._raise_error_if_profile_exists()
             except ProfileAttributesEmptyError:
                 if self.raise_error_on_empty_attributes:
@@ -551,6 +549,34 @@ def generate_synthetic_data(
                 keyword_parameters=keyword_parameters,
             )
 
+    def translate(
+        self, text: str, source_language: str, target_language: str
+    ) -> Union[str, None]:
+        """
+        Translate a text using a source language and a target language
+
+        :param str text: Text to translate
+        :param str source_language: Source language
+        :param str target_language: Target language
+        :return: str
+        """
+        parameters = {
+            "profile_name": self.profile_name,
+            "text": text,
+            "source_language": source_language,
+            "target_language": target_language,
+        }
+        with cursor() as cr:
+            data = cr.callfunc(
+                "DBMS_CLOUD_AI.TRANSLATE",
+                oracledb.DB_TYPE_CLOB,
+                keyword_parameters=parameters,
+            )
+        if data is not None:
+            result = data.read()
+            return result
+        return None
+
 
 class Session:
     """Session lets you persist request parameters across DBMS_CLOUD_AI

tests/profiles/test_1200_profile.py

@@ -79,11 +79,9 @@ def python_gen_ai_neg_feedback(cursor, python_gen_ai_profile):
         ],
     )
     cursor.execute(
-        f"""
-                  BEGIN
-                      dbms_cloud_ai.set_profile('{python_gen_ai_profile.profile_name}');
-                  END;
-       """
+        f"""BEGIN
+             dbms_cloud_ai.set_profile('{python_gen_ai_profile.profile_name}');
+            END;"""
     )
     prompt = "Total points of each gymnasts"
     action = select_ai.Action.SHOWSQL
@@ -113,10 +111,9 @@ def python_gen_ai_pos_feedback(cursor, python_gen_ai_profile):
         ["prompt", "action", "sql_text"],
     )
     cursor.execute(
-        f"""
-                  BEGIN
-                      dbms_cloud_ai.set_profile('{python_gen_ai_profile.profile_name}');
-                  END;
+        f"""BEGIN
+             dbms_cloud_ai.set_profile('{python_gen_ai_profile.profile_name}');
+            END;
        """
     )
     prompt = "Lists the name of all people"
@@ -349,3 +346,11 @@ def test_1217(cursor, python_gen_ai_profile, python_gen_ai_pos_feedback):
     assert (
         feedback_attributes["sql_text"] == python_gen_ai_pos_feedback.sql_text
     )
+
+
+def test_1218(python_gen_ai_profile):
+    """Test translate"""
+    response = python_gen_ai_profile.translate(
+        text="Thank you", source_language="en", target_language="de"
+    )
+    assert response == "Danke"

tests/profiles/test_1300_profile_async.py

@@ -83,10 +83,9 @@ async def python_gen_ai_neg_feedback(async_cursor, python_gen_ai_profile):
         ],
     )
     await async_cursor.execute(
-        f"""
-                  BEGIN
-                      dbms_cloud_ai.set_profile('{python_gen_ai_profile.profile_name}');
-                  END;
+        f"""BEGIN
+             dbms_cloud_ai.set_profile('{python_gen_ai_profile.profile_name}');
+            END;
        """
     )
     prompt = "Total points of each gymnasts"
@@ -117,10 +116,9 @@ async def python_gen_ai_pos_feedback(async_cursor, python_gen_ai_profile):
         ["prompt", "action", "sql_text"],
     )
     await async_cursor.execute(
-        f"""
-                  BEGIN
-                      dbms_cloud_ai.set_profile('{python_gen_ai_profile.profile_name}');
-                  END;
+        f"""BEGIN
+             dbms_cloud_ai.set_profile('{python_gen_ai_profile.profile_name}');
+            END;
        """
     )
     prompt = "Lists the name of all people"
@@ -370,3 +368,11 @@ async def test_1317(
     assert (
         feedback_attributes["sql_text"] == python_gen_ai_pos_feedback.sql_text
     )
+
+
+async def test_1318(python_gen_ai_profile):
+    """Test translate"""
+    response = await python_gen_ai_profile.translate(
+        text="Thank you", source_language="en", target_language="de"
+    )
+    assert response == "Danke"