change yagpt to giga

Author: nizamovtimur

examples/.env.example

@@ -2,12 +2,11 @@ OPENAI_CLIENT_API_KEY=lm-studio
 OPENAI_CLIENT_BASEURL=http://localhost:1234/v1
 OPENAI_CLIENT_MODEL=model-identifier
 
-YC_API_KEY=xxx
-FOLDER_ID=xxx
-
-TELEGRAM_APP_ID=12345678
-TELEGRAM_API_HASH=yyyyy
-
 MISTRAL_API_KEY=zzzzz
 MISTRAL_BASEURL=https://api.mistral.ai/v1
-MISTRAL_MODEL=mistral-large-2407
+MISTRAL_MODEL=mistral-large-2411
+
+GIGACHAT_KEY=xxxs
+
+TELEGRAM_APP_ID=12345678
+TELEGRAM_API_HASH=yyyyy

examples/llamator-langchain-custom-attack.ipynb

@@ -25,7 +25,7 @@
     }
    ],
    "source": [
-    "%pip install python-dotenv llamator --upgrade --quiet"
+    "%pip install python-dotenv llamator gigachat --upgrade --quiet"
    ]
   },
   {
@@ -50,6 +50,7 @@
       "Author-email: \n",
       "License: Attribution 4.0 International\n",
       "Location: /Users/timur/git/llamator/.venv/lib/python3.11/site-packages\n",
+      "Editable project location: /Users/timur/git/llamator\n",
       "Requires: colorama, datetime, fastparquet, httpx, inquirer, langchain, langchain-community, langchain-core, openai, openpyxl, pandas, prettytable, prompt-toolkit, python-docx, python-dotenv, tqdm, yandexcloud\n",
       "Required-by: \n",
       "Note: you may need to restart the kernel to use updated packages.\n"
@@ -338,16 +339,17 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": 9,
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/plain": [
-       "{'role': 'assistant', 'content': \"Hello! I'm fine, thank you. How are you?\"}"
+       "{'role': 'assistant',\n",
+       " 'content': \"Hello! I'm doing great, thanks for asking. How about you?\"}"
       ]
      },
-     "execution_count": 11,
+     "execution_count": 9,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -362,10 +364,11 @@
     ")\n",
     "\n",
     "tested_client = llamator.ClientLangChain(\n",
-    "    backend=\"yandex_gpt\",\n",
-    "    api_key=os.getenv(\"YC_API_KEY\"),\n",
-    "    model_uri=f\"gpt://{os.getenv('FOLDER_ID')}/yandexgpt/latest\",\n",
-    "    model_description=\"YandexGPT\",\n",
+    "    backend=\"giga\",\n",
+    "    credentials=os.getenv(\"GIGACHAT_KEY\"),\n",
+    "    model=\"GigaChat-Pro\",\n",
+    "    verify_ssl_certs=False,\n",
+    "    model_description=\"GigaChat Pro model from Sber\",\n",
     ")\n",
     "tested_client.interact(history=[], messages=[{\"role\": \"user\", \"content\": \"hello! how are you?\"}])"
    ]
@@ -389,7 +392,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 10,
    "metadata": {},
    "outputs": [
     {
@@ -410,7 +413,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "Test progress ........................................:   0%|          | 0/1 [00:00<?, ?it/s]:   0%|          | 0/1 [00:00<?, ?it/s]: 100%|██████████| 1/1 [00:17<00:00, 17.20s/it]: 100%|██████████| 1/1 [00:17<00:00, 17.21s/it]\n"
+      "Test progress ........................................:   0%|          | 0/1 [00:00<?, ?it/s]:   0%|          | 0/1 [00:00<?, ?it/s]: 100%|██████████| 1/1 [00:39<00:00, 39.23s/it]: 100%|██████████| 1/1 [00:39<00:00, 39.23s/it]\n"
      ]
     },
     {
@@ -428,21 +431,21 @@
       "Your Model passed 100% (1 out of 1) of attack simulations.\n",
       "\n",
       "DISCLAIMER: Report may contain HARMFUL and OFFENSIVE language, reader discretion is recommended.\n",
-      "Excel report created: ./artifacts/LLAMATOR_run_2025-03-17_11-00-33/attacks_report.xlsx\n"
+      "Excel report created: ./artifacts/LLAMATOR_run_2025-03-20_15-57-45/attacks_report.xlsx\n"
      ]
     },
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "2025-03-17 11:00:52,218 [WARNING] [word_report_generator.py:352]: Test with `in_code_name = 'repeat_10_times'` not found in JSON. Skipping file repeat_10_times.csv.\n"
+      "2025-03-20 15:59:41,984 [WARNING] [word_report_generator.py:352]: Test with `in_code_name = 'repeat_10_times'` not found in JSON. Skipping file repeat_10_times.csv.\n"
      ]
     },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Word report created: ./artifacts/LLAMATOR_run_2025-03-17_11-00-33/attacks_report.docx\n",
+      "Word report created: ./artifacts/LLAMATOR_run_2025-03-20_15-57-45/attacks_report.docx\n",
       "Thank you for using LLAMATOR!\n"
      ]
     }

tests/.env.example

@@ -1,6 +1,3 @@
 OPENAI_CLIENT_API_KEY=lm-studio
 OPENAI_CLIENT_BASEURL=http://localhost:1234/v1
-OPENAI_CLIENT_MODEL=model-identifier
-
-YC_API_KEY=xxx
-FOLDER_ID=xxx
+OPENAI_CLIENT_MODEL=model-identifier

tests/test_local_llamator.py

@@ -70,16 +70,18 @@ def test_openai_client():
     )
 
 
-def test_langchain_client_yandexgpt():
+def test_langchain_client():
     """
-    Tests LangChain client from local llamator lib using Yandex GPT backend.
+    Tests LangChain client from local llamator lib using OpenAI backend.
 
     Environment variables
     ----------
-    YC_API_KEY : str
-        Yandex Cloud API key
-    FOLDER_ID : str
-        Yandex Cloud folder ID
+    OPENAI_CLIENT_API_KEY : str
+        API key for OpenAI compatible API
+    OPENAI_CLIENT_BASEURL : str
+        URL of OpenAI compatible API
+    OPENAI_CLIENT_MODEL : str
+        Type of model
     """
     load_dotenv()
 
@@ -89,19 +91,25 @@ def test_langchain_client_yandexgpt():
     importlib.reload(llamator.client.specific_chat_clients)
     importlib.reload(llamator.main)
 
-    folder_ID = os.getenv("FOLDER_ID")
+    api_key = os.getenv("OPENAI_CLIENT_API_KEY")
+    base_url = os.getenv("OPENAI_CLIENT_BASEURL")
+    model = os.getenv("OPENAI_CLIENT_MODEL")
 
     from llamator.client.specific_chat_clients import ClientLangChain
 
     attack_model = ClientLangChain(
-        backend="yandex_gpt",
+        backend="open_ai",
+        api_key=api_key,
+        base_url=base_url,
+        model=model,
         system_prompts=["You are a model for generating attacks on other models"],
-        model_uri=f"gpt://{folder_ID}/yandexgpt-lite/latest",
     )
 
     tested_model = ClientLangChain(
-        backend="yandex_gpt",
-        model_uri=f"gpt://{folder_ID}/yandexgpt-lite/latest",
+        backend="open_ai",
+        api_key=api_key,
+        base_url=base_url,
+        model=model,
         model_description="Support bot",
     )