shen/Snap-Solver
1
0
Fork 0
mirror of https://github.com/Zippland/Snap-Solver.git synced 2026-01-19 09:41:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix bugs

Browse Source
This commit is contained in:
ChengHao
2025-05-08 20:38:19 +08:00
parent 0393afd9df
commit 3c663b2839
3 changed files with 38 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
models/base.py
View File

@@ -2,11 +2,12 @@ from abc import ABC, abstractmethod
from typing import Generator, Any
class BaseModel(ABC):
def __init__(self, api_key: str, temperature: float = 0.7, system_prompt: str = None, language: str = None):
def __init__(self, api_key: str, temperature: float = 0.7, system_prompt: str = None, language: str = None, api_base_url: str = None):
self.api_key = api_key
self.temperature = temperature
self.language = language
self.system_prompt = system_prompt or self.get_default_system_prompt()
self.api_base_url = api_base_url
@abstractmethod
def analyze_image(self, image_data: str, proxies: dict = None) -> Generator[dict, None, None]: