mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-08-08 03:44:14 +00:00
fix: another way
This commit is contained in:
parent
24e6d16a09
commit
76fdf52b0a
@ -639,7 +639,11 @@ def _setup_fastapi(worker_params: ModelWorkerParameters, app=None):
|
||||
router as controller_router,
|
||||
)
|
||||
|
||||
if not worker_params.controller_addr and worker_params.model_name != "proxyllm":
|
||||
if not worker_params.controller_addr:
|
||||
# if we have http_proxy or https_proxy in env, the server can not start
|
||||
# so set it to empty here
|
||||
os.environ["http_proxy"] = ""
|
||||
os.environ["https_proxy"] = ""
|
||||
worker_params.controller_addr = f"http://127.0.0.1:{worker_params.port}"
|
||||
logger.info(
|
||||
f"Run WorkerManager with standalone mode, controller_addr: {worker_params.controller_addr}"
|
||||
|
@ -1,5 +1,6 @@
|
||||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
import os
|
||||
from dataclasses import dataclass, field
|
||||
from enum import Enum
|
||||
from typing import Dict, Optional
|
||||
@ -246,6 +247,11 @@ class ProxyModelParameters(BaseModelParameters):
|
||||
proxy_api_key: str = field(
|
||||
metadata={"tags": "privacy", "help": "The api key of current proxy LLM"},
|
||||
)
|
||||
http_proxy: Optional[str] = field(
|
||||
default=os.environ.get("http_proxy") or os.environ.get("https_proxy"),
|
||||
metadata={"help": "The http or https proxy to use openai"},
|
||||
)
|
||||
|
||||
proxyllm_backend: Optional[str] = field(
|
||||
default=None,
|
||||
metadata={
|
||||
|
Loading…
Reference in New Issue
Block a user