Feat/chatglm3 function call (#197)

dataelement · Dec 17, 2023 · 9e8e412 · 9e8e412
2 parents 4a388e5 + 7b5b4d8
commit 9e8e412
Show file tree

Hide file tree

Showing 25 changed files with 1,521 additions and 12 deletions.
diff --git a/src/backend/Dockerfile b/src/backend/Dockerfile
@@ -11,7 +11,7 @@ RUN curl -sSL https://install.python-poetry.org | python3 -
 # # Add Poetry to PATH
 ENV PATH="${PATH}:/root/.local/bin"
 # # Copy the pyproject.toml and poetry.lock files
-COPY poetry.lock pyproject.toml ./
+# COPY poetry.lock pyproject.toml ./
 # Copy the rest of the application codes
 COPY ./ ./
 
@@ -22,4 +22,4 @@ RUN python -m pip install --upgrade pip && \
 RUN poetry config virtualenvs.create false
 RUN poetry install --no-interaction --no-ansi --without dev
 
-CMD ["uvicorn", "bisheng.main:app", "--workers", "2",  "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "bisheng.main:app", "--workers", "2",  "--host", "0.0.0.0", "--port", "7860"]
diff --git a/src/backend/pyproject.toml b/src/backend/pyproject.toml
@@ -18,7 +18,7 @@ include = ["./bisheng/*", "bisheng/**/*"]
 bisheng = "bisheng.__main__:main"
 
 [tool.poetry.dependencies]
-bisheng_langchain = "v0.1.8"
+bisheng_langchain = "0.2.0rc1"
 bisheng_pyautogen = "0.1.18"
 minio = "^7.2.0"
 fastapi_jwt_auth = "^0.5.0"

diff --git a/src/bisheng-langchain/bisheng_langchain/agents/__init__.py b/src/bisheng-langchain/bisheng_langchain/agents/__init__.py
@@ -0,0 +1,5 @@
+from bisheng_langchain.agents.chatglm_functions_agent.base import ChatglmFunctionsAgent
+
+__all__ = [
+    ChatglmFunctionsAgent
+]
diff --git a/src/bisheng-langchain/bisheng_langchain/agents/chatglm_functions_agent/__init__.py b/src/bisheng-langchain/bisheng_langchain/agents/chatglm_functions_agent/__init__.py