Setting DB URL

Dockerfile

@@ -1,6 +1,6 @@
 FROM python:3.11-slim
 WORKDIR /app
 COPY . .
-RUN pip install flask praisonai==1.0.1 gunicorn markdown
+RUN pip install flask praisonai==1.0.2 gunicorn markdown
 EXPOSE 8080
 CMD ["gunicorn", "-b", "0.0.0.0:8080", "api:app"]

docs/api/praisonai/deploy.html

@@ -110,7 +110,7 @@ Loads environment variables from .env file or system and sets them.</p>
             file.write(&#34;FROM python:3.11-slim\n&#34;)
             file.write(&#34;WORKDIR /app\n&#34;)
             file.write(&#34;COPY . .\n&#34;)
-            file.write(&#34;RUN pip install flask praisonai==1.0.1 gunicorn markdown\n&#34;)
+            file.write(&#34;RUN pip install flask praisonai==1.0.2 gunicorn markdown\n&#34;)
             file.write(&#34;EXPOSE 8080\n&#34;)
             file.write(&#39;CMD [&#34;gunicorn&#34;, &#34;-b&#34;, &#34;0.0.0.0:8080&#34;, &#34;api:app&#34;]\n&#39;)
             

praisonai.rb

@@ -3,7 +3,7 @@ class Praisonai < Formula
   
     desc "AI tools for various AI applications"
     homepage "https://github.com/MervinPraison/PraisonAI"
-    url "https://github.com/MervinPraison/PraisonAI/archive/refs/tags/1.0.1.tar.gz"
+    url "https://github.com/MervinPraison/PraisonAI/archive/refs/tags/1.0.2.tar.gz"
     sha256 "1828fb9227d10f991522c3f24f061943a254b667196b40b1a3e4a54a8d30ce32"  # Replace with actual SHA256 checksum
     license "MIT"
   

praisonai/cli.py

@@ -26,7 +26,9 @@ CREWAI_AVAILABLE = False
 AUTOGEN_AVAILABLE = False
 
 try:
-    os.environ["CHAINLIT_APP_ROOT"] = os.path.join(os.path.expanduser("~"), ".praison")
+    # Set CHAINLIT_APP_ROOT only if it doesn't exist
+    if "CHAINLIT_APP_ROOT" not in os.environ:
+        os.environ["CHAINLIT_APP_ROOT"] = os.path.join(os.path.expanduser("~"), ".praison")
     from chainlit.cli import chainlit_run
     CHAINLIT_AVAILABLE = True
 except ImportError:
@@ -389,7 +391,8 @@ class PraisonAI:
             import praisonai
             os.environ["CHAINLIT_PORT"] = "8084"
             root_path = os.path.join(os.path.expanduser("~"), ".praison")
-            os.environ["CHAINLIT_APP_ROOT"] = root_path
+            if "CHAINLIT_APP_ROOT" not in os.environ:
+                os.environ["CHAINLIT_APP_ROOT"] = root_path
             public_folder = os.path.join(os.path.dirname(praisonai.__file__), 'public')
             if not os.path.exists(os.path.join(root_path, "public")):
                 if os.path.exists(public_folder):
@@ -412,7 +415,8 @@ class PraisonAI:
             import praisonai
             os.environ["CHAINLIT_PORT"] = "8086"
             root_path = os.path.join(os.path.expanduser("~"), ".praison")
-            os.environ["CHAINLIT_APP_ROOT"] = root_path
+            if "CHAINLIT_APP_ROOT" not in os.environ:
+                os.environ["CHAINLIT_APP_ROOT"] = root_path
             public_folder = os.path.join(os.path.dirname(__file__), 'public')
             if not os.path.exists(os.path.join(root_path, "public")):
                 if os.path.exists(public_folder):
@@ -481,7 +485,8 @@ class PraisonAI:
             import praisonai
             os.environ["CHAINLIT_PORT"] = "8088"
             root_path = os.path.join(os.path.expanduser("~"), ".praison")
-            os.environ["CHAINLIT_APP_ROOT"] = root_path
+            if "CHAINLIT_APP_ROOT" not in os.environ:
+                os.environ["CHAINLIT_APP_ROOT"] = root_path
             public_folder = os.path.join(os.path.dirname(praisonai.__file__), 'public')
             if not os.path.exists(os.path.join(root_path, "public")):
                 if os.path.exists(public_folder):

praisonai/deploy.py

@@ -56,7 +56,7 @@ class CloudDeployer:
             file.write("FROM python:3.11-slim\n")
             file.write("WORKDIR /app\n")
             file.write("COPY . .\n")
-            file.write("RUN pip install flask praisonai==1.0.1 gunicorn markdown\n")
+            file.write("RUN pip install flask praisonai==1.0.2 gunicorn markdown\n")
             file.write("EXPOSE 8080\n")
             file.write('CMD ["gunicorn", "-b", "0.0.0.0:8080", "api:app"]\n')
             

praisonai/ui/chat.py

@@ -175,7 +175,12 @@ initialize_db()
 
 deleted_thread_ids = []  # type: List[str]
 
-cl_data._data_layer = SQLAlchemyDataLayer(conninfo=f"sqlite+aiosqlite:///{DB_PATH}")
+# Get database connection string from environment variable or use SQLite as default
+DB_URL = os.getenv("DATABASE_URL", f"sqlite+aiosqlite:///{DB_PATH}")
+cl_data._data_layer = SQLAlchemyDataLayer(
+    conninfo=DB_URL,
+    ssl_require=bool(os.getenv("DATABASE_SSL", False))
+)
 
 # Set Tavily API key
 tavily_api_key = os.getenv("TAVILY_API_KEY")

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "PraisonAI"
-version = "1.0.1"
+version = "1.0.2"
 description = "PraisonAI application combines AutoGen and CrewAI or similar frameworks into a low-code solution for building and managing multi-agent LLM systems, focusing on simplicity, customization, and efficient human-agent collaboration."
 authors = ["Mervin Praison"]
 license = ""
@@ -48,6 +48,7 @@ pyngrok = {version = ">=1.4.0", optional = true}
 sqlalchemy = {version = ">=2.0.36", optional = true}
 playwright = {version = ">=1.47.0", optional = true}
 openai = {version = ">=1.54.0", optional = true}
+pydantic = {version = "<=2.10.1", optional = true}
 
 [tool.poetry.group.docs.dependencies]
 mkdocs = "*"
@@ -125,7 +126,10 @@ chat = [
     "tavily-python", 
     "crawl4ai",
     "sqlalchemy",
-    "playwright"
+    "playwright",
+    "rich",
+    "instructor",
+    "pydantic"
 ]
 code = [
     "chainlit", 
@@ -135,7 +139,8 @@ code = [
     "tavily-python", 
     "crawl4ai",
     "sqlalchemy",
-    "playwright"
+    "playwright",
+    "pydantic"
 ]
 train = ["setup-conda-env"]
 realtime = [