Fix main pypi workflow (#157)

.github/workflows/upload-pypi-dev.yml

@@ -83,4 +83,5 @@ jobs:
 
       - name: Push Docker image to Docker Hub
         run: |
-          docker push tensoropsai/llmstudio:${{ env.VERSION }}
+          docker push tensoropsai/llmstudio:${{ env.VERSION }}
+          

.github/workflows/upload-pypi.yml

@@ -84,4 +84,5 @@ jobs:
       - name: Push Docker images to Docker Hub
         run: |
           docker push tensoropsai/llmstudio:${{ env.VERSION }}
-          docker push tensoropsai/llmstudio:latest
+          docker push tensoropsai/llmstudio:latest
+

Dockerfile

@@ -0,0 +1,15 @@
+FROM python:3.11-slim
+ENV PYTHONUNBUFFERED=1
+
+# Install tools
+RUN apt-get clean && apt-get update
+
+# Install llmstudio
+ARG LLMSTUDIO_VERSION
+RUN pip install llmstudio==${LLMSTUDIO_VERSION}
+RUN pip install psycopg2-binary
+
+# Expose Ports
+EXPOSE 8001 8002
+
+CMD ["llmstudio", "server"]

llmstudio/engine/providers/openai.py

@@ -41,6 +41,7 @@ async def generate_client(
         """Generate an OpenAI client"""
 
         try:
+            # Create client
             client = OpenAI(api_key=request.api_key or self.API_KEY)
             return await asyncio.to_thread(
                 client.chat.completions.create,