langchain support (#5)

* ✨ Refactor main classes and functions * add langchain and langsmith support
franperezlopez · Jan 2, 2024 · 16688b0 · 16688b0
1 parent 8c3cd9f
commit 16688b0
Show file tree

Hide file tree

Showing 9 changed files with 677 additions and 311 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -17,8 +17,8 @@ ARG AZURE_OPENAI_API_KEY
 ENV AZURE_OPENAI_API_KEY=$AZURE_OPENAI_API_KEY
 ARG AZURE_OPENAI_API_BASE
 ENV AZURE_OPENAI_API_BASE=$AZURE_OPENAI_API_BASE
-ARG LANGCHAIN_API_KEY
-ENV LANGCHAIN_API_KEY=$LANGCHAIN_API_KEY
+ARG LANGSMITH_API_KEY
+ENV LANGSMITH_API_KEY=$LANGSMITH_API_KEY
 
 
 # Copy environment file
@@ -35,8 +35,7 @@ WORKDIR /app
 
 # Copy source code
 COPY src src
-COPY bot.py .
 
 EXPOSE 80
 
-CMD ["python", "-m", "bot"]
+CMD ["python", "-m", "src.bot"]
diff --git a/bot.py b/bot.py
diff --git a/environment.yml b/environment.yml
@@ -11,7 +11,10 @@ dependencies:
   - pillow=10.1.0
   - pip
   - pip:
+    - langchain[openai]==0.0.353
     - loguru==0.7.2
     - piexif==1.1.3
     - python-telegram-bot[webhooks]==20.7
     - google-search-results==2.4.2
+    - randomname==0.2.1
+
diff --git a/requirements_dev.txt b/requirements_dev.txt
@@ -0,0 +1,6 @@
+ipykernel
+ruff
+pytest
+pytest-cov
+pytest-moq
+mypy