diff --git a/src/process.py b/src/process.py index 21adb15..66d1aa8 100644 --- a/src/process.py +++ b/src/process.py @@ -11,7 +11,7 @@ class SiloconFlowAPI: self.url = base_url self.count = 0 def get_ai_message(self, message,maxcount,i=0): - with OpenAI(api_key=self.api_key, base_url=self.base_url) as client: + with OpenAI(api_key=self.api_key, base_url=self.url) as client: try: response = client.chat.completions.create( model=self.module[i],