casbin
/
casibase
1
0
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Rename to "OpenAI" type

This commit is contained in:
Yang Luo 2023-09-08 01:14:53 +08:00
parent 3a27f56531
commit a80a385bd3
5 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
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
}

8
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()

2
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>)
}

2
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: "",

2
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"},