github/k8sgpt
1
0
Fork 0
mirror of https://github.com/k8sgpt-ai/k8sgpt.git synced 2025-08-14 22:13:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: add providerId to serve mode (#1260)

Browse Source 
Signed-off-by: yanweili <yanweili@ibm.com>
Co-authored-by: yanweili <yanweili@ibm.com>
Signed-off-by: AlexsJones <alexsimonjones@gmail.com>
This commit is contained in:
Yanwei Li 2024-09-26 23:32:58 -07:00 committed by AlexsJones
parent b1528c77a0
commit ff73434d34
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
cmd/serve/serve.go
View File

@ -14,10 +14,11 @@ limitations under the License.
package serve package serve
import ( import (
k8sgptserver "github.com/k8sgpt-ai/k8sgpt/pkg/server"
"os" "os"
"strconv" "strconv"
k8sgptserver "github.com/k8sgpt-ai/k8sgpt/pkg/server"
"github.com/fatih/color" "github.com/fatih/color"
"github.com/k8sgpt-ai/k8sgpt/pkg/ai" "github.com/k8sgpt-ai/k8sgpt/pkg/ai"
"github.com/spf13/cobra" "github.com/spf13/cobra"
@ -108,6 +109,7 @@ var ServeCmd = &cobra.Command{
baseURL := os.Getenv("K8SGPT_BASEURL") baseURL := os.Getenv("K8SGPT_BASEURL")
engine := os.Getenv("K8SGPT_ENGINE") engine := os.Getenv("K8SGPT_ENGINE")
proxyEndpoint := os.Getenv("K8SGPT_PROXY_ENDPOINT") proxyEndpoint := os.Getenv("K8SGPT_PROXY_ENDPOINT")
providerId := os.Getenv("K8SGPT_PROVIDER_ID")
// If the envs are set, allocate in place to the aiProvider // If the envs are set, allocate in place to the aiProvider
// else exit with error // else exit with error
envIsSet := backend != "" || password != "" || model != "" envIsSet := backend != "" || password != "" || model != ""
@ -119,6 +121,7 @@ var ServeCmd = &cobra.Command{
BaseURL: baseURL, BaseURL: baseURL,
Engine: engine, Engine: engine,
ProxyEndpoint: proxyEndpoint, ProxyEndpoint: proxyEndpoint,
ProviderId: providerId,
Temperature: temperature(), Temperature: temperature(),
TopP: topP(), TopP: topP(),
TopK: topK(), TopK: topK(),