FreshPixels commited on
Commit
40af5f2
·
verified ·
1 Parent(s): 5ae94cb

Upload interpreter_config.py

Browse files
Files changed (1) hide show
  1. interpreter_config.py +25 -0
interpreter_config.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """Конфигурация Open Interpreter для моделей"""
2
+ import logging
3
+ from typing import Optional
4
+
5
+ def configure_interpreter_for_model(model_name: str, state, api_key: str, api_base: str) -> bool:
6
+ """
7
+ Настраивает Open Interpreter для работы с заданной моделью.
8
+ Возвращает True при успешной настройке, False в случае ошибки.
9
+ """
10
+ try:
11
+ from interpreter import interpreter
12
+ model = state.models.get(model_name, state.models["deepseek-v4-pro"])
13
+ interpreter.llm.model = f"openai/{model.endpoint}"
14
+ interpreter.llm.api_key = api_key
15
+ interpreter.llm.api_base = api_base
16
+ interpreter.llm.context_window = model.context_window
17
+ interpreter.llm.max_tokens = model.max_tokens
18
+ logging.info(f"Open Interpreter configured for model: {model_name}")
19
+ return True
20
+ except ImportError as e:
21
+ logging.error(f"Open Interpreter not available: {e}")
22
+ return False
23
+ except Exception as e:
24
+ logging.error(f"Failed to configure interpreter for {model_name}: {e}")
25
+ return False