release 0.0.5

This commit is contained in:
Adil Hafeez 2024-10-29 16:35:20 -07:00
parent bb9a774a72
commit 8904d28590
6 changed files with 35 additions and 17 deletions

View file

@ -58,6 +58,8 @@ def main(ctx, version):
click.echo(f"archgw cli version: {get_version()}") click.echo(f"archgw cli version: {get_version()}")
ctx.exit() ctx.exit()
log.info(f"Starting archgw cli version: {get_version()}")
if ctx.invoked_subcommand is None: if ctx.invoked_subcommand is None:
click.echo("""Arch (The Intelligent Prompt Gateway) CLI""") click.echo("""Arch (The Intelligent Prompt Gateway) CLI""")
click.echo(logo) click.echo(logo)
@ -68,7 +70,7 @@ def main(ctx, version):
@click.option( @click.option(
"--service", "--service",
default=SERVICE_ALL, default=SERVICE_ALL,
help="Optioanl parameter to specify which service to build. Options are model_server, archgw", help="Optional parameter to specify which service to build. Options are model_server, archgw",
) )
def build(service): def build(service):
"""Build Arch from source. Must be in root of cloned repo.""" """Build Arch from source. Must be in root of cloned repo."""
@ -178,6 +180,12 @@ def up(file, path, service):
env = os.environ.copy() env = os.environ.copy()
# check if access_keys are preesnt in the config file # check if access_keys are preesnt in the config file
access_keys = get_llm_provider_access_keys(arch_config_file=arch_config_file) access_keys = get_llm_provider_access_keys(arch_config_file=arch_config_file)
# remove duplicates
access_keys = set(access_keys)
# remove the $ from the access_keys
access_keys = [item[1:] if item.startswith("$") else item for item in access_keys]
if access_keys: if access_keys:
if file: if file:
app_env_file = os.path.join( app_env_file = os.path.join(
@ -186,6 +194,7 @@ def up(file, path, service):
else: else:
app_env_file = os.path.abspath(os.path.join(path, ".env")) app_env_file = os.path.abspath(os.path.join(path, ".env"))
print(f"app_env_file: {app_env_file}")
if not os.path.exists( if not os.path.exists(
app_env_file app_env_file
): # check to see if the environment variables in the current environment or not ): # check to see if the environment variables in the current environment or not

16
arch/tools/poetry.lock generated
View file

@ -13,7 +13,7 @@ files = [
[[package]] [[package]]
name = "archgw_modelserver" name = "archgw_modelserver"
version = "0.0.4" version = "0.0.5"
description = "A model server for serving models" description = "A model server for serving models"
optional = false optional = false
python-versions = "*" python-versions = "*"
@ -250,13 +250,13 @@ tqdm = ["tqdm"]
[[package]] [[package]]
name = "huggingface-hub" name = "huggingface-hub"
version = "0.26.1" version = "0.26.2"
description = "Client library to download and publish models, datasets and other repos on the huggingface.co hub" description = "Client library to download and publish models, datasets and other repos on the huggingface.co hub"
optional = false optional = false
python-versions = ">=3.8.0" python-versions = ">=3.8.0"
files = [ files = [
{file = "huggingface_hub-0.26.1-py3-none-any.whl", hash = "sha256:5927a8fc64ae68859cd954b7cc29d1c8390a5e15caba6d3d349c973be8fdacf3"}, {file = "huggingface_hub-0.26.2-py3-none-any.whl", hash = "sha256:98c2a5a8e786c7b2cb6fdeb2740893cba4d53e312572ed3d8afafda65b128c46"},
{file = "huggingface_hub-0.26.1.tar.gz", hash = "sha256:414c0d9b769eecc86c70f9d939d0f48bb28e8461dd1130021542eff0212db890"}, {file = "huggingface_hub-0.26.2.tar.gz", hash = "sha256:b100d853465d965733964d123939ba287da60a547087783ddff8a323f340332b"},
] ]
[package.dependencies] [package.dependencies]
@ -785,13 +785,13 @@ type = ["importlib-metadata (>=7.0.2)", "jaraco.develop (>=7.21)", "mypy (==1.11
[[package]] [[package]]
name = "tqdm" name = "tqdm"
version = "4.66.5" version = "4.66.6"
description = "Fast, Extensible Progress Meter" description = "Fast, Extensible Progress Meter"
optional = false optional = false
python-versions = ">=3.7" python-versions = ">=3.7"
files = [ files = [
{file = "tqdm-4.66.5-py3-none-any.whl", hash = "sha256:90279a3770753eafc9194a0364852159802111925aa30eb3f9d85b0e805ac7cd"}, {file = "tqdm-4.66.6-py3-none-any.whl", hash = "sha256:223e8b5359c2efc4b30555531f09e9f2f3589bcd7fdd389271191031b49b7a63"},
{file = "tqdm-4.66.5.tar.gz", hash = "sha256:e1020aef2e5096702d8a025ac7d16b1577279c9d63f8375b63083e9a5f0fcbad"}, {file = "tqdm-4.66.6.tar.gz", hash = "sha256:4bdd694238bef1485ce839d67967ab50af8f9272aab687c0d7702a01da0be090"},
] ]
[package.dependencies] [package.dependencies]
@ -834,4 +834,4 @@ zstd = ["zstandard (>=0.18.0)"]
[metadata] [metadata]
lock-version = "2.0" lock-version = "2.0"
python-versions = "^3.10" python-versions = "^3.10"
content-hash = "e51783523cbe087cb1db94e874a28564b43af03a1689523d3738b212e288f64b" content-hash = "f3210810e843b529258ea0ac42681503a19f693f5d63ca6bb281485a828006de"

View file

@ -23,7 +23,7 @@ click = "^8.1.7"
jinja2 = "^3.1.4" jinja2 = "^3.1.4"
jsonschema = "^4.23.0" jsonschema = "^4.23.0"
setuptools = "75.2.0" setuptools = "75.2.0"
archgw_modelserver = "0.0.4" archgw_modelserver = "0.0.5"
huggingface_hub = "^0.26.0" huggingface_hub = "^0.26.0"
[tool.poetry.scripts] [tool.poetry.scripts]

View file

@ -16,10 +16,6 @@
"name": "model_server", "name": "model_server",
"path": "model_server" "path": "model_server"
}, },
{
"name": "chatbot_ui",
"path": "chatbot_ui"
},
{ {
"name": "e2e_tests", "name": "e2e_tests",
"path": "e2e_tests" "path": "e2e_tests"

View file

@ -1,3 +1,4 @@
import importlib
import sys import sys
import os import os
import time import time
@ -7,6 +8,15 @@ import tempfile
import subprocess import subprocess
import logging import logging
def get_version():
try:
version = importlib.metadata.version("archgw_modelserver")
return version
except importlib.metadata.PackageNotFoundError:
return "version not found"
logging.basicConfig( logging.basicConfig(
level=logging.INFO, level=logging.INFO,
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
@ -15,6 +25,8 @@ logging.basicConfig(
log = logging.getLogger("model_server.cli") log = logging.getLogger("model_server.cli")
log.setLevel(logging.INFO) log.setLevel(logging.INFO)
log.info(f"model server version: {get_version()}")
def run_server(port=51000): def run_server(port=51000):
"""Start, stop, or restart the Uvicorn server based on command-line arguments.""" """Start, stop, or restart the Uvicorn server based on command-line arguments."""
@ -37,8 +49,9 @@ def run_server(port=51000):
def start_server(port=51000): def start_server(port=51000):
"""Start the Uvicorn server""" """Start the Uvicorn server"""
log.info( log.info(
"Starting model server - loading some awesomeness, this may take some time :)" "starting model server - loading some awesomeness, this may take some time :)"
) )
process = subprocess.Popen( process = subprocess.Popen(
[ [
"python", "python",
@ -61,7 +74,7 @@ def start_server(port=51000):
log.info(f"Model server started with PID {process.pid}") log.info(f"Model server started with PID {process.pid}")
else: else:
# Add model_server boot-up logs # Add model_server boot-up logs
log.info("Model server - Didn't Sart In Time. Shutting Down") log.info("model server - didn't start in time, shutting down")
process.terminate() process.terminate()

View file

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "archgw_modelserver" name = "archgw_modelserver"
version = "0.0.4" version = "0.0.5"
description = "A model server for serving models" description = "A model server for serving models"
authors = ["Katanemo Labs, Inc <archgw@katanemo.com>"] authors = ["Katanemo Labs, Inc <archgw@katanemo.com>"]
license = "Apache 2.0" license = "Apache 2.0"