45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from typing import Any
|
|
from AnP.Controllers.AIController import AIController
|
|
from AnP.Drivers.WebSocketServerDriver import WebSocketServerDriver
|
|
from AnP.Drivers.HTTPSocketServerDriver import HTTPSocketServerDriver
|
|
from AnP.Drivers.HTTPServerDriver import HTTPServerDriver
|
|
from AnP.Drivers.OllamaDriver import OllamaDriver
|
|
from AnP.Drivers.FilesDriver import FilesDriver
|
|
from AnP.Drivers.XMLScrapDriver import XMLScrapDriver
|
|
from AnP.Drivers.SeleniumDriver import SeleniumDriver
|
|
from AnP.Drivers.SQLiteDriver import SQLiteDriver
|
|
from AnP.Controllers.CloneController import CloneController
|
|
from AnP.Controllers.AITranslateController import AITranslateController
|
|
|
|
inputs:dict[str, dict[str, Any|None]] = {
|
|
"default_models" : {
|
|
"AIController" : AIController,
|
|
"CloneController" : CloneController,
|
|
"AITranslateController" : AITranslateController,
|
|
"WebSocketServerDriver" : WebSocketServerDriver,
|
|
"HTTPSocketServerDriver" : HTTPSocketServerDriver,
|
|
"HTTPServerDriver" : HTTPServerDriver,
|
|
"OllamaDriver" : OllamaDriver,
|
|
"FilesDriver" : FilesDriver,
|
|
"XMLScrapDriver" : XMLScrapDriver,
|
|
"SeleniumDriver" : SeleniumDriver,
|
|
"SQLiteDriver" : SQLiteDriver
|
|
}
|
|
}
|
|
|
|
try:
|
|
|
|
from secrets import secrets as custom_secrets
|
|
|
|
for key, value in dict(custom_secrets).items():
|
|
if key not in inputs or isinstance(inputs[key], dict):
|
|
inputs[key] = value
|
|
elif isinstance(value, dict):
|
|
for subkey, subvalue in value.items():
|
|
inputs[key][subkey] = subvalue
|
|
|
|
except ImportError:
|
|
pass |