Merge pull request #10 from epam/fixes-from-development

feat: fixes from development

Author: Oleksii-Klimov

.github/workflows/pr_check_tests.yml

@@ -8,7 +8,7 @@ on:
 
 jobs:
   run_tests:
-    uses: epam/ai-dial-ci/.github/workflows/test_python_docker.yml@0.1.0
+    uses: epam/ai-dial-ci/.github/workflows/test_python_docker.yml@0.3.0
     with:
       bypass_checks: false
       python_version: 3.11

.github/workflows/release.yml

@@ -9,5 +9,5 @@ env:
 
 jobs:
   release:
-    uses: epam/ai-dial-ci/.github/workflows/publish_python_docker.yml@0.1.0
+    uses: epam/ai-dial-ci/.github/workflows/publish_python_docker.yml@0.3.0
     secrets: inherit

aidial_assistant/application/__init__.py

@@ -0,0 +1,3 @@
+import logging
+
+logger = logging.getLogger(__name__)

aidial_assistant/application/assistant_application.py

@@ -1,11 +1,13 @@
 from pathlib import Path
 
+from aidial_sdk import HTTPException
 from aidial_sdk.chat_completion.base import ChatCompletion
 from aidial_sdk.chat_completion.request import Addon, Request
 from aidial_sdk.chat_completion.response import Response
 from aiohttp import hdrs
-from openai import InvalidRequestError
+from openai import InvalidRequestError, OpenAIError
 
+from aidial_assistant.application import logger
 from aidial_assistant.application.args import parse_args
 from aidial_assistant.application.prompts import (
     MAIN_SYSTEM_DIALOG_MESSAGE,
@@ -112,7 +114,16 @@ async def chat_completion(
         )
         with response.create_single_choice() as choice:
             callback = ServerChainCallback(choice)
-            await chain.run_chat(history, callback, usage_publisher)
+            try:
+                await chain.run_chat(history, callback, usage_publisher)
+            except OpenAIError as e:
+                logger.exception("Request processing has failed.")
+                raise HTTPException(
+                    str(e),
+                    status_code=e.http_status or 500,
+                    code=e.code,
+                )
+
             choice.set_state(callback.state)
 
         response.set_usage(

aidial_assistant/json_stream/parsing_context.py

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "aidial-assistant"
-version = "0.1.0rc"
+version = "0.1.0"
 description = "DIAL Assistant Service"
 authors = ["EPAM RAIL <SpecialEPM-DIALDevTeam@epam.com>"]
 homepage = "https://epam-rail.com"
@@ -19,7 +19,7 @@ clean = "scripts.clean:main"
 python = "^3.11"
 aiocache = "^0.12.2"
 jinja2 = "^3.1.2"
-langchain = "^0.0.309"
+langchain = "^0.0.312"
 openai = "^0.28.0"
 pydantic = "1.10.13"
 pyyaml = "^6.0.1"