Update ai_model to be more generic #410

	name: Benchmarks Workflow

	on:
	issue_comment:
	types: [created]

	jobs:
	build-and-test:
	if: github.event.issue.pull_request && contains(github.event.comment.body, '/benchmark')
	runs-on: ubuntu-latest

	services:
	postgres:
	image: pgvector/pgvector:pg16
	env:
	POSTGRES_HOST: localhost
	POSTGRES_USER: ${{ secrets.DB_USER }}
	PGUSER: ${{ secrets.DB_USER }}
	POSTGRES_PASSWORD: ${{ secrets.DB_PASS }}
	POSTGRES_DB: ${{ secrets.DB_NAME }}
	ports:
	- 5432:5432
	options: >-
	--health-cmd pg_isready
	--health-interval 10s
	--health-timeout 5s
	--health-retries 5

	steps:
	- uses: actions/checkout@v3

	- name: Set Environment and Install Dependencies
	uses: ./.github/actions/setup-env

	- name: Run server
	run: \|
	./run serve &
	env:
	OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
	DB_USER: ${{ secrets.DB_USER }}
	DB_PASS: ${{ secrets.DB_PASS }}
	DB_NAME: ${{ secrets.DB_NAME }}
	GIT_TOKEN: ${{ secrets.GIT_TOKEN }}
	USER_DB_ADMIN: ${{ secrets.USER_DB_ADMIN }}
	USER_DB_PASS: ${{ secrets.USER_DB_PASS }}
	USER_DB_HOST: ${{ secrets.USER_DB_HOST }}
	DB_PORT: 5432
	RUN_ENV: local
	PORT: 8080

	- name: Run Tests
	run: \|
	./run benchmark

	- name: Clean up
	if: always()
	run: \|
	lsof -ti:8080 \| grep LISTEN \| xargs --no-run-if-empty kill -9

	env:
	DB_USER: ${{ secrets.DB_USER }}
	DB_PASS: ${{ secrets.DB_PASS }}
	DB_NAME: ${{ secrets.DB_NAME }}
	DB_PORT: 5432
	OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
	RUN_ENV: local
	PORT: 8080
	DATABASE_URL: postgresql://${{ secrets.DB_USER }}:${{ secrets.DB_PASS }}@localhost:5432/${{ secrets.DB_NAME }}

Provide feedback