Reorged folders.

.gitignore

@@ -3,3 +3,5 @@
 __pycache__
 .env
 config.yml.debug
+debug.yml
+dist

Dockerfile

@@ -1,25 +1,20 @@
 # Use an official Python runtime as the base image
-FROM python:3.9
+FROM python:3.10
 
 # Set the working directory in the container
 WORKDIR /app
 
 # Copy all files from the current directory to the working directory in the container
-COPY . /app/
+COPY config.yml poetry.lock pyproject.toml /app/
+COPY demo /app/demo/
+COPY meta_prompt /app/meta_prompt/
 
-# Install the required dependencies from requirements.txt
-RUN pip install --no-cache-dir -r requirements.txt
-
-# Set the environment variables
-ENV API_KEY=""
-ENV PROXY=""
-ENV OPENAI_API_BASE=""
-ENV OTHER_ARGS="--advanced_mode"
-ENV ADVANCED_MODE="true"
-ENV SERVER_NAME="0.0.0.0"
+RUN pip install --no-cache-dir -U poetry
+RUN poetry config virtualenvs.create false
+RUN poetry install --with=dev
 
 # Expose the port (if necessary)
 EXPOSE 7860
 
 # Run the script when the container launches
-CMD /usr/local/bin/python3 meta_prompt.py --api_key=${API_KEY} --proxy=${PROXY} --openai_api_base=${OPENAI_API_BASE} --server_name=${SERVER_NAME} ${OTHER_ARGS}
+CMD python demo/gradio_meta_prompt.py

{demo → app}/config.py

@@ -12,5 +12,7 @@ class LLMConfig(BaseModel):
 class MetaPromptConfig(BaseConfig):
     llms: Optional[dict[str, LLMConfig]]
     examples_path: Optional[str]
-    server_name: Optional[str] = '127.0.0.1'
-    server_port: Optional[int] = 7878
+    server_name: Optional[str] = None
+    server_port: Optional[int] = None
+    recursion_limit: Optional[int] = 25
+    recursion_limit_max: Optional[int] = 50

{demo → app}/gradio_meta_prompt.py

@@ -2,7 +2,7 @@ import gradio as gr
 from confz import BaseConfig, CLArgSource, EnvSource, FileSource
 from meta_prompt import MetaPromptGraph, AgentState
 from langchain_openai import ChatOpenAI
-from config import MetaPromptConfig
+from app.config import MetaPromptConfig
 
 class LLMModelFactory:
     def __init__(self):
@@ -85,8 +85,8 @@ iface = gr.Interface(
     ],
     additional_inputs=[
         gr.Textbox(label="Initial System Message", show_copy_button=True, value=""),
-        gr.Number(label="Recursion Limit", value=25,
-                  precision=0, minimum=1, maximum=100, step=1),
+        gr.Number(label="Recursion Limit", value=config.recursion_limit,
+                  precision=0, minimum=1, maximum=config.recursion_limit_max, step=1),
         gr.Dropdown(
             label="Model Name",
             choices=config.llms.keys(),

config.yml

@@ -1,4 +1,12 @@
 llms:
+  groq/llama3-70b-8192:
+    type: ChatOpenAI
+    temperature: 0.1
+    model_name: "llama3-70b-8192"
+    openai_api_key: ""
+    openai_api_base: "https://api.groq.com/openai/v1"
+    max_tokens: 8192
+    verbose: true
   anthropic/claude-3-haiku:
     type: ChatOpenAI
     temperature: 0.1
@@ -23,15 +31,7 @@ llms:
     openai_api_base: "https://openrouter.ai/api/v1"
     max_tokens: 8192
     verbose: true
-  groq/llama3-70b-8192:
-    type: ChatOpenAI
-    temperature: 0.1
-    model_name: "llama3-70b-8192"
-    openai_api_key: ""
-    openai_api_base: "https://api.groq.com/openai/v1"
-    max_tokens: 8192
-    verbose: true
 
-examples_path: "demo/examples"
+examples_path: "app/examples"
 server_name: 0.0.0.0
-server_port: 7870
+server_port: 7860

meta_prompt/__init__.py

@@ -0,0 +1,4 @@
+__version__ = '0.1.0'
+
+from .meta_prompt import AgentState, MetaPromptGraph
+

pyproject.toml

@@ -0,0 +1,24 @@
+[tool.poetry]
+name = "meta-prompt"
+version = "0.1.0"
+description = "This package generates high quality prompts with input and expected output examples."
+authors = ["Yale Huang <[email protected]>"]
+license = "MIT"
+
+[tool.poetry.dependencies]
+python = "^3.10"
+langgraph = "^0.1.5"
+langchain = "^0.2.6"
+langchain-openai = "^0.1.14"
+pydantic = "^2.8.2"
+
+[tool.poetry.dev-dependencies]
+gradio = "^4.37.2"
+confz = "^2.0.1"
+
+[tool.poetry.group.dev.dependencies]
+pytest = "^8.2.2"
+
+[build-system]
+requires = ["poetry-core>=1.0.0"]
+build-backend = "poetry.core.masonry.api"

src/meta_prompt/__init__.py

@@ -1 +0,0 @@
-from .meta_prompt import AgentState, MetaPromptGraph

tests/meta_prompt_graph_test.py

@@ -79,9 +79,9 @@ class TestMetaPromptGraph(unittest.TestCase):
         assert updated_state.accepted == True
 
     def test_workflow_execution(self):
-        MODEL_NAME = "anthropic/claude-3.5-sonnet:haiku"
+        # MODEL_NAME = "anthropic/claude-3.5-sonnet:beta"
         # MODEL_NAME = "meta-llama/llama-3-70b-instruct"
-        # MODEL_NAME = "deepseek/deepseek-chat"
+        MODEL_NAME = "deepseek/deepseek-chat"
         # MODEL_NAME = "google/gemma-2-9b-it"
         # MODEL_NAME = "recursal/eagle-7b"
         # MODEL_NAME = "meta-llama/llama-3-8b-instruct"
@@ -116,7 +116,7 @@ class TestMetaPromptGraph(unittest.TestCase):
         print(result.content)
 
     def test_workflow_execution_with_llms(self):
-        optimizer_llm = ChatOpenAI(model_name="anthropic/claude-3.5-sonnet:haiku", temperature=0.5)
+        optimizer_llm = ChatOpenAI(model_name="deepseek/deepseek-chat", temperature=0.5)
         executor_llm = ChatOpenAI(model_name="meta-llama/llama-3-8b-instruct", temperature=0.01)
 
         llms = {