File size: 1,151 Bytes
011bd45
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
import os
import logging
from openai import OpenAI




try:
    from dotenv import load_dotenv
    load_dotenv()
except ImportError:
    pass


class Envs:
    def __init__(self):
        self.hf_token = os.getenv("HF_TOKEN")
        self.api_key = os.getenv("API_KEY")
        self.end_task = os.getenv("END_TASK")
        self.end_lang = os.getenv("END_LANG")
        self.model_agent = os.getenv("MODEL_AGENT")
        self.model_mllm = os.getenv("MODEL_MLLM")
        self.model_loc = os.getenv("MODEL_LOC")
        
        # Only initialize OpenAI client if we have the required env vars
        if self.end_lang and self.api_key:
            self.client = OpenAI(base_url=self.end_lang, api_key=self.api_key)
        else:
            self.client = None
            print("WARNING: OpenAI client not initialized due to missing environment variables")
            
        self.debug = os.getenv("DEBUG", "1").lower() in ("true", "1", "yes")
        self.fps = int(os.getenv("FPS"))
    

env = Envs()

logger = logging.getLogger('copilot')
logger.setLevel(logging.DEBUG if env.debug else logging.INFO)
logger.addHandler(logging.StreamHandler())