BrandAnalysis/agent/glm.py

from config import load_config
from zhipuai import ZhipuAI
class Glm:
    _instance = None
    def __new__(cls):
        if not cls._instance:
            cls._instance = super(Glm, cls).__new__(cls)
            cls._instance._initialized = False
        return cls._instance
    
    def __init__(self):
        if not self._initialized:
            self.cfg = load_config()['glm']
            self.client = ZhipuAI(api_key=self.cfg["api_key"])
            self._initialized = True
            
        self.model_name = "glm-4-plus"
    
    def text_response(self, query):
        resonse = self.client.chat.completions.create(
            model=self.model_name,
            messages=[
                {"role": "user", "content": query}
            ],
            response_format= {
                'type': 'json_object'
            }
        )
        return resonse.choices[0].message
    
    def set_modelname(self, modelname):
        self.model_name = modelname
    
if __name__ == '__main__':
    glm = Glm()
    response = glm.response("请帮我编写一段快速排序的代码")
    print(response)