fix/azure openai (#346)

* Fix Azure OpenAI embedding startup issue.

* add new openai models

* cleanup

config.yaml

@@ -16,6 +16,8 @@ llm:
   # embeddings deployment is required when Zep is configured to use OpenAI embeddings
   #   embedding_deployment: "text-embedding-ada-002-customname"
   # Use only with an alternate OpenAI-compatible API endpoint
+    llm_deployment:
+    embedding_deployment:
   openai_endpoint:
   openai_org_id:
 nlp:

docker-compose.dev.yaml

@@ -1,4 +1,3 @@
-version: "3.7"
 services:
   db:
     image: ghcr.io/getzep/postgres:latest

docker-compose.yaml

@@ -1,10 +1,9 @@
-version: "3.7"
 services:
   db:
     image: ghcr.io/getzep/postgres:latest
     container_name: zep-postgres
     restart: on-failure
-    shm_size: "128mb"  # Increase this if vacuuming fails with a "no space left on device" error
+    shm_size: "128mb" # Increase this if vacuuming fails with a "no space left on device" error
     environment:
       - POSTGRES_USER=postgres
       - POSTGRES_PASSWORD=postgres
@@ -16,7 +15,7 @@ services:
     volumes:
       - zep-db:/var/lib/postgresql/data
     healthcheck:
-      test: [ "CMD", "pg_isready", "-q", "-d", "postgres", "-U", "postgres" ]
+      test: ["CMD", "pg_isready", "-q", "-d", "postgres", "-U", "postgres"]
       interval: 5s
       timeout: 5s
       retries: 5
@@ -39,12 +38,12 @@ services:
     container_name: zep
     restart: on-failure
     depends_on:
-        db:
-          condition: service_healthy
-        nlp:
-          condition: service_healthy
+      db:
+        condition: service_healthy
+      nlp:
+        condition: service_healthy
     ports:
-        - "8000:8000"
+      - "8000:8000"
     volumes:
       - ./config.yaml:/app/config.yaml
     environment:

pkg/llms/llm_base.go

@@ -158,6 +158,8 @@ var ValidOpenAILLMs = map[string]bool{
 	"gpt-3.5-turbo-1106": true,
 	"gpt-4-32k":          true,
 	"gpt-4-1106-preview": true,
+	"gpt-4-turbo":        true,
+	"gpt-4o":             true,
 }
 
 var ValidAnthropicLLMs = map[string]bool{
@@ -174,6 +176,8 @@ var MaxLLMTokensMap = map[string]int{
 	"gpt-4":              8192,
 	"gpt-4-32k":          32_768,
 	"gpt-4-1106-preview": 128_000,
+	"gpt-4-turbo":        128_000,
+	"gpt-4o":             128_000,
 	"claude-instant-1":   100_000,
 	"claude-2":           100_000,
 }

pkg/testutils/data.go

@@ -1,5 +1,3 @@
-//go:build testutils
-
 package testutils
 
 import "github.com/getzep/zep/pkg/models"

pkg/testutils/utils.go

@@ -1,5 +1,3 @@
-//go:build testutils
-
 package testutils
 
 import (