update: req and unit_config.json

aimclub · Jan 19, 2025 · a989347 · a989347
1 parent 7410afb
commit a989347
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 3 deletions.
diff --git a/protollm_tools/llm-api/protollm_api/backend/endpoints.py b/protollm_tools/llm-api/protollm_api/backend/endpoints.py
@@ -10,8 +10,6 @@
 from protollm_sdk.object_interface.redis_wrapper import RedisWrapper
 from protollm_sdk.object_interface.rabbit_mq_wrapper import RabbitMQWrapper
 
-from tests.integration.test_local_RMQ import rabbitmq_connection
-
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 

diff --git a/protollm_tools/llm-api/requirements.txt b/protollm_tools/llm-api/requirements.txt
@@ -52,7 +52,7 @@ pika==1.3.2 ; python_version >= "3.10" and python_version < "4.0"
 prometheus-client==0.21.0 ; python_version >= "3.10" and python_version < "4.0"
 prompt-toolkit==3.0.48 ; python_version >= "3.10" and python_version < "4.0"
 propcache==0.2.0 ; python_version >= "3.10" and python_version < "4.0"
-protollm-sdk==1.0.0 ; python_version >= "3.10" and python_version < "4.0"
+protollm-sdk==1.1.0 ; python_version >= "3.10" and python_version < "4.0"
 pydantic-core==2.23.4 ; python_version >= "3.10" and python_version < "4.0"
 pydantic==2.9.2 ; python_version >= "3.10" and python_version < "4.0"
 pygments==2.18.0 ; python_version >= "3.10" and python_version < "4.0"

diff --git a/protollm_tools/llm-api/unit_config.json b/protollm_tools/llm-api/unit_config.json
@@ -0,0 +1,15 @@
+{
+    "listeners": {
+        "*:6672": {
+            "pass": "applications/backend"
+        }
+    },
+    "applications": {
+        "backend": {
+            "type": "python",
+            "path": ".",
+            "module": "protollm_api.backend.main",
+            "callable": "app"
+        }
+    }
+}