diff --git a/sgpt/handlers/handler.py b/sgpt/handlers/handler.py index ab03fd47..60c2ad51 100644 --- a/sgpt/handlers/handler.py +++ b/sgpt/handlers/handler.py @@ -11,11 +11,14 @@ completion: Callable[..., Any] = lambda *args, **kwargs: Generator[Any, None, None] base_url = cfg.get("API_BASE_URL") use_litellm = cfg.get("USE_LITELLM") == "true" +user = cfg.get("OPEN_AI_USER") additional_kwargs = { "timeout": int(cfg.get("REQUEST_TIMEOUT")), "api_key": cfg.get("OPENAI_API_KEY"), "base_url": None if base_url == "default" else base_url, } +if user and user != "": + additional_kwargs["user"] = user if use_litellm: import litellm # type: ignore