Skip to content

ryanchaiyakul/socratic_models

Repository files navigation

Socratic Models

A flexible framework for multi agent conversations.

Setup

  1. Create a "secret.py" file with a "SECRET_KEY" variable for your Gemini API key.
  2. Create a python virtual environment and install the necessary packages.
python -m venv .venv
source .venv/bin/activate               # Or .venv/Scripts/activate for windows
pip install -r requirements.txt

CLI

Call 'main.py' with the numbers of requested turns (each model uses a turn so 2 turns for both models to speak once).

py main.py 4

Web Server (WIP)

alt text

Call the following command in the repositories root directory. The web server is currently extremely barebones and will be developed into a full application for possibly human testing.

flask run

Current Features

  • A new session can be created by entering a prompt on a new page
  • You can load previous sessions by entering the UUID at the top of the page
  • Clicking the advance button will cause the web server to generate the next response
  • You can add an intermediary statement by writing it in the textbox and hitting enter
  • The full dialogue will be output in raw HTML

Improvements

  • Add OpenAI actor class
  • Add hyperlink from overview page to specific sessions for quality of life

About

A multi agent conversation framework

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published