Merge pull request #2 from trentmillar/feat/client_config

feature, openai clientconfig handles azure openai

Author: geekr-dev

.gitignore

@@ -1,2 +1,6 @@
+# ide
+.vscode/
+.idea/
+
 example/memory
 example/workspace

README.md

@@ -20,19 +20,26 @@ GPT Engineer is made to be easy to adapt, extend, and make your agent learn how
 
 ## Usage
 
-**Setup**:
+### Setup:
 
+**OpenAI**
 - `go mod tidy`
 - `export OPENAI_API_KEY=[your api key]` with a key that has GPT4 access, if you don't have GPT4 access, then it will use GPT-3.5 as a fallback.
 
-**Run**:
+**Azure**
+- `export OPENAI_BASE=[your Azure OpenAI endpoint]` should be _https://[deployment name].openai.azure.com/_
+
+### Run:
 - Edit `example/main_prompt` to specify what you want to build
 - Run `go run .`, default language is English, if you want to use other language, and `-lang` argument like `go run . -lang=Chinese`
 
-**Results**:
+**(optional) Azure**
+- use the argument `-model [your deployment name]`
+
+### Results:
 - Check the generated files in `example/workspace`
 
-### Limitations
+## Limitations
 
 Implementing additional chain of thought prompting, e.g. [Reflexion](https://github.com/noahshinn024/reflexion), should be able to make it more reliable and not miss requested functionality in the main prompt.
 

ai.go

@@ -24,9 +24,23 @@ type AI struct {
 }
 
 func NewAI(model string, temperature float64, lang string) *AI {
-	client := openai.NewClient(os.Getenv("OPENAI_API_KEY"))
+	var clientConfig openai.ClientConfig
+	apiKey := os.Getenv("OPENAI_API_KEY")
+	apiBase := os.Getenv("OPENAI_API_BASE")
+	if apiBase == "" {
+		clientConfig = openai.DefaultConfig(apiKey)
+	} else {
+		clientConfig = openai.DefaultAzureConfig(apiKey, apiBase)
+	}
+	client := openai.NewClientWithConfig(clientConfig)
 	ai := &AI{model: model, client: client, temperature: temperature, lang: lang}
-	_, err := client.GetModel(context.Background(), "gpt-4")
+
+	// Azure uses deployments and is not exposed in the openai api so model is assumed to be okay
+	if clientConfig.APIType == openai.APITypeAzure {
+		return ai
+	}
+
+	_, err := client.GetModel(context.Background(), model)
 	if err != nil {
 		fmt.Println("Model gpt-4 not available for provided api key reverting to gpt-3.5.turbo. Sign up for the gpt-4 wait list here: https://openai.com/waitlist/gpt-4-api")
 		ai.model = "gpt-3.5-turbo"