Browse Source

Rename to "OpenAI" type

HEAD
Yang Luo 2 years ago
parent
commit
a80a385bd3
5 changed files with 9 additions and 9 deletions
  1. +2
    -2
      ai/model.go
  2. +4
    -4
      ai/openai.go
  3. +1
    -1
      web/src/ProviderEditPage.js
  4. +1
    -1
      web/src/ProviderListPage.js
  5. +1
    -1
      web/src/Setting.js

+ 2
- 2
ai/model.go View File

@@ -24,8 +24,8 @@ type ModelProvider interface {
}

func GetModelProvider(typ string, subType string, secretKey string) (ModelProvider, error) {
if typ == "OpenAI API" {
p, err := NewOpenaiGpt3p5ModelProvider(subType, secretKey)
if typ == "OpenAI" {
p, err := NewOpenAiModelProvider(subType, secretKey)
if err != nil {
return nil, err
}


+ 4
- 4
ai/openai.go View File

@@ -24,20 +24,20 @@ import (
"github.com/sashabaranov/go-openai"
)

type OpenaiGpt3p5ModelProvider struct {
type OpenAiModelProvider struct {
SubType string
SecretKey string
}

func NewOpenaiGpt3p5ModelProvider(subType string, secretKey string) (*OpenaiGpt3p5ModelProvider, error) {
p := &OpenaiGpt3p5ModelProvider{
func NewOpenAiModelProvider(subType string, secretKey string) (*OpenAiModelProvider, error) {
p := &OpenAiModelProvider{
SubType: subType,
SecretKey: secretKey,
}
return p, nil
}

func (p *OpenaiGpt3p5ModelProvider) QueryText(question string, writer io.Writer, builder *strings.Builder) error {
func (p *OpenAiModelProvider) QueryText(question string, writer io.Writer, builder *strings.Builder) error {
client := getProxyClientFromToken(p.SecretKey)

ctx := context.Background()


+ 1
- 1
web/src/ProviderEditPage.js View File

@@ -117,7 +117,7 @@ class ProviderEditPage extends React.Component {
<Select virtual={false} style={{width: "100%"}} value={this.state.provider.type} onChange={(value => {this.updateProviderField("type", value);})}>
{
[
{id: "OpenAI API", name: "OpenAI API"},
{id: "OpenAI", name: "OpenAI"},
{id: "Hugging Face", name: "Hugging Face"},
].map((item, index) => <Option key={index} value={item.id}>{item.name}</Option>)
}


+ 1
- 1
web/src/ProviderListPage.js View File

@@ -54,7 +54,7 @@ class ProviderListPage extends React.Component {
createdTime: moment().format(),
displayName: `New Provider - ${randomName}`,
category: "Model",
type: "OpenAI API",
type: "OpenAI",
subType: "gpt-3.5-turbo",
clientId: "",
clientSecret: "",


+ 1
- 1
web/src/Setting.js View File

@@ -654,7 +654,7 @@ export function isResponseDenied(data) {
}
export function getProviderSubTypeOptions(type) {
if (type === "OpenAI API") {
if (type === "OpenAI") {
return (
[
{id: "gpt-4-32k-0613", name: "gpt-4-32k-0613"},


Loading…
Cancel
Save