diff --git a/config/config.yaml b/config/config.yaml index 2f747b7..40fd8b6 100644 --- a/config/config.yaml +++ b/config/config.yaml @@ -6,6 +6,7 @@ models: endpoint: https://api.openai.com/v1/chat/completions auth_env_var: OPENAI_API_KEY org_env_var: OPENAI_ORG_ID + project_env_var: OPENAI_PROJECT_ID prompt: [ { @@ -20,6 +21,7 @@ models: endpoint: https://api.openai.com/v1/chat/completions auth_env_var: OPENAI_API_KEY org_env_var: OPENAI_ORG_ID + project_env_var: OPENAI_PROJECT_ID prompt: [ { @@ -34,6 +36,7 @@ models: endpoint: https://api.openai.com/v1/chat/completions auth_env_var: OPENAI_API_KEY org_env_var: OPENAI_ORG_ID + project_env_var: OPENAI_PROJECT_ID prompt: [ { diff --git a/llm/llm.go b/llm/llm.go index 35e43e5..ee3839f 100644 --- a/llm/llm.go +++ b/llm/llm.go @@ -48,6 +48,9 @@ func (c *LLMClient) createRequest(payload Payload) (*http.Request, error) { if c.config.OrgID != "" { req.Header.Set("OpenAI-Organization", c.config.OrgID) } + if c.config.ProjectID != "" { + req.Header.Set("OpenAI-Project", c.config.ProjectID) + } req.Header.Set("Content-Type", "application/json") return req, nil } diff --git a/types/types.go b/types/types.go index 525fb94..1f0c88b 100644 --- a/types/types.go +++ b/types/types.go @@ -5,6 +5,7 @@ type ModelConfig struct { Endpoint string `yaml:"endpoint"` Auth string `yaml:"auth_env_var"` OrgID string `yaml:"org_env_var,omitempty"` + ProjectID string `yaml:"project_env_var,omitempty"` Prompt []Message `yaml:"prompt"` }