update

bak/get_weather/.env.example

@@ -1,4 +0,0 @@
-SUPABASE_URL=https://bbvyxjvleoizsipxydsm.supabase.co
-SUPABASE_KEY=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6ImJidnl4anZsZW9penNpcHh5ZHNtIiwicm9sZSI6ImFub24iLCJpYXQiOjE3NDQ1MzkwNjIsImV4cCI6MjA2MDExNTA2Mn0.cbUNhiCHXqIaW4leU_vGTKVI51gLVvsFOVWYS_vJrA4
-GOOGLE_GENAI_USE_VERTEXAI=FALSE
-GOOGLE_API_KEY=AIzaSyBYPwMLibOIoM-SQHuWjZ-V4LLmmMxqhm8

bak/get_weather/__init__.py

@@ -1 +0,0 @@
-from . import agent 

bak/get_weather/agent.py

@@ -1,53 +0,0 @@
-### adk web ./teams
-from google.adk.agents import Agent
-from .tools.get_current_time import get_current_time
-from .tools.get_weather import get_weather
-from .cache import get_api_key
-# from .team import init_team, get_team
-import os
-
-def my_before_model_callback(callback_context, **kwargs):
-    os.environ["GOOGLE_API_KEY"] = get_api_key()  # 直接设置环境变量
-print('888888888888888888888888888888888888888888888')
-
-greeting_agent = Agent(
-    name="greeting_agent",
-    model="gemini-2.5-flash-preview-04-17",
-    instruction="You are the Greeting Agent. Your ONLY task is to provide a friendly greeting to the user. "
-                "Do not engage in any other conversation or tasks."
-                "before you answer, you should say your name: 'I am greeting_agent'",
-    description="Handles simple greetings and hellos",
-    before_model_callback=my_before_model_callback
-)
-
-farewell_agent = Agent(
-    name="farewell_agent",
-    model="gemini-2.5-flash-preview-04-17",
-    instruction="You are the Farewell Agent. Your ONLY task is to provide a polite goodbye message. "
-                "If not about goodbye, just transfer to weather_time_agent"
-                "Do not perform any other actions."
-                "before you answer, you should say your name: 'I am farewell_agent'",
-    before_model_callback=my_before_model_callback,
-    description="Handles simple farewells and goodbyes",
-)
-
-print("\n\n\n\n\n\n\n\n&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&\nroot_agent init")
-root_agent=Agent(
-    name="weather_time_agent",
-    model="gemini-2.5-flash-preview-04-17",
-    description=(
-        "Agent to answer questions about the time and weather in a city."
-    ),
-    instruction=(
-        "You are a helpful agent who can answer user questions about the time and weather in a city."
-        "before you answer, you should say your name: 'I am weather_time_agent'"
-    ),
-    # global_instruction = '请在每一句回答里都加上：“我的主人”',
-    tools=[get_weather, get_current_time], 
-    sub_agents=[greeting_agent, farewell_agent], 
-    before_model_callback=my_before_model_callback
-)
-
-import pprint
-print('\n\n\n\n*************************************\n\nroot_agent dict: \n')
-pprint.pprint(root_agent.__dict__) # 尝试打印对象的字典表示

bak/get_weather/readme.md

@@ -1,19 +0,0 @@
-## 默认约定
-
-### 目录结构
-
-ADK-TEAMS          # 项目根目录
-    |- teams             # 包含多个团队的目录代码，每一个子目录为一个团队
-    |    |- get_weather  # 团队目录。示例：获取天气信息团队
-    |        |- agents     # agents 目录。处理从数据库加载 agents 的代码
-    |        |- tools     # tools 目录。处理从数据库加载 tools 的代码
-    |        |- __init__.py    # 初始化代码
-    |        |- .env    # 环境变量文件
-    |        |- agent.py    # 团队的 agent 入口代码
-    |        |- readme.md    # 团队的 README 文件（当前文件）
-    |        |- tools.py    # 团队的 tools 入口代码
-    |- .gitattributes #  Git 属性文件
-    |- .gitignore    #  Git 忽略文件
-    |- Dockerfile    #  Dockerfile 文件
-    |- README.md    # 项目的 README 文件
-    |- requirements.txt    # 项目的依赖文件

bak/get_weather/team.py

@@ -1,33 +0,0 @@
-import sys, os
-sys.path.append(os.path.abspath(os.path.join(os.path.dirname(os.path.abspath(__file__)), '..', '..')))
-from team_framework.load_team_config import load_team_config
-from team_framework.load_tools import load_tools
-from team_framework.load_agents import load_agents, load_root_agent
-from google.adk.agents import Agent
-
-_team = {}
-
-def init_team():
-    global _team 
-    _team= {}
-    _team_config = load_team_config()
-    print('\n\n\n\n\n***************************\n_team_config')
-    print(_team_config)
-    _team["tools"] = load_tools(_team_config['tools'])
-    _team["agents"] = load_agents(_team_config['agents'])
-
-    _team["root_agent"] = load_root_agent(_team_config['root_agent'])
-    _team["root_agent"].tools = [] ## 为 root_agent 添加工具
-    for item in _team_config['root_agent']['tools']:
-        # print('\n\n\n\n|||||||||||||||||| item',item)
-        _team["root_agent"].tools.append(_team['tools'][item])
-    _team["root_agent"].sub_agents=[]
-    for item in _team_config['root_agent']['sub_agents']:
-        print('\n\n\n\n|||||||||||||||||||||||||||||||||||||||||||||||||||||||||||| \nitem',item)
-        print(_team['agents'][item])
-        _team["root_agent"].sub_agents.append(_team['agents'][item])
-        # _team["root_agent"].agents.append(_team['agents'][item])
-
-def get_team():
-    global _team
-    return _team

bak/get_weather/tool.py.copy

@@ -1,23 +0,0 @@
-import importlib
-
-def load_tools_from_array(arr:list):
-    ret = {}
-    for item in arr:
-        # 构建正确的模块路径，使用 __name__ 动态获取当前模块的父路径（此处正好为团队名称），并加上 .tools 子目录和工具名称
-        parent_module_path = '.'.join(__name__.split('.')[:-1])
-        module_path = f'{parent_module_path}.tools.{item}'
-        try:
-            module = importlib.import_module(module_path)
-            # 函数名预期与工具名称相同
-            func = getattr(module, item)
-            ret[item] = func
-            print(f"成功从模块 '{module_path}' 导入函数 '{item}'")
-        except ImportError as e:
-            print(f"导入模块 {module_path} 时出错: {e}")
-        except AttributeError as e:
-            print(f"在模块 {module_path} 中找不到函数 '{item}': {e}")
-    return ret
-
-def load_tools_from_db():
-    tool_arr = ["get_weather", "get_current_time"]
-    return load_tools_from_array(tool_arr)

teams/ai_quant/cache.py

@@ -1,104 +0,0 @@
-import os
-from supabase import create_client, Client
-import time
-
-# Simple in-memory cache
-api_key_cache = {}
-last_loaded_time = 0
-CACHE_TTL = 300 # Cache time-to-live in seconds (5 minutes)
-
-def get_supabase_client() -> Client:
-    """Initializes and returns a Supabase client."""
-    url: str = os.environ.get("SUPABASE_URL")
-    key: str = os.environ.get("SUPABASE_KEY")
-    if not url or not key:
-        print("Supabase URL or Key not found in environment variables.")
-        return None
-    return create_client(url, key)
-
-def load_api_keys_from_db():
-    """Loads API keys from the database into the cache."""
-    global api_key_cache
-
-    print("Loading API keys from database...")
-    supabase: Client = get_supabase_client()
-    if not supabase:
-        return {}
-
-    try:
-        response = supabase.from_("app_config").select("config").eq("id", 1).execute()
-        #[
-        #   {
-        #       'config': {
-        #           'api_keys': [
-        #               {'api_key': 'xxxxx', 'can_use_at': 0}, 
-        #               {'api_key': 'yyyyy', 'can_use_at': 0}
-        #           ]
-        #       }
-        #   }
-        #]
-        # print(f"Supabase response data: {response.data}")
-
-        if response.data and response.data[0] and 'config' in response.data[0] and 'api_keys' in response.data[0]['config']:
-            api_keys_list = response.data[0]['config']['api_keys']
-            if api_keys_list:
-                api_key_cache = {item['api_key']: item['can_use_at'] for item in api_keys_list}
-                # {'xxxxx': 0, 'yyyyy': 0}
-                # print(f"Successfully loaded {len(api_key_cache)} API keys with can_use_at timestamps.")
-                # print(f"API keys cache: {api_key_cache}")
-            else:
-                print("API keys list is empty in the database response.")
-                api_key_cache = {}
-        else:
-            print("Unexpected database response structure or no data found.")
-            api_key_cache = {}
-
-    except Exception as e:
-        print(f"Error loading API keys from database: {e}")
-        api_key_cache = {} # Clear cache on error
-
-    return api_key_cache
-
-def get_api_key():
-    """Gets an API key from the cache, loading from DB if necessary."""
-    global api_key_cache
-    # print("\n\n\n\n\nGetting API key...")
-    # print(api_key_cache)
-
-    # Ensure cache is loaded
-    if not api_key_cache :
-        load_api_keys_from_db()
-    # api_key_cache={'xxxxx': 1, 'yyyyy': 0}
-    # print('api_key_cache',api_key_cache)
-    
-    # Sort available keys by their 'can_use_at' timestamp in ascending order
-    sorted_key_cache = sorted(api_key_cache, key=lambda key: api_key_cache.get(key, float('inf')))
-    # print('\n\n\n\n\n\n\n\n\nsorted_key_cache',sorted_key_cache)
-
-    # Select the key with the smallest 'can_use_at' value
-    selected_key = sorted_key_cache[0]
-
-    # TODO: Update the 'can_use_at' timestamp for the selected key in the cache
-    # This requires knowing the rate limit or delay period for the API.
-    # For now, we'll just return the key. In a real system, you'd update the cache
-    # with a future timestamp after using the key successfully.
-    # Example: api_key_cache[selected_key] = time.time() + API_RATE_LIMIT_DELAY
-
-    # print(f"Using API key: {selected_key[:5]}...") # Print partial key for security
-    api_key_cache[selected_key]=time.time()
-    return selected_key
-
-# Example usage (for testing)
-if __name__ == "__main__":
-    # In a real application, environment variables would be set
-    # For testing, you might set them here or use a .env file loader
-    # os.environ['SUPABASE_URL'] = 'your_supabase_url'
-    # os.environ['SUPABASE_KEY'] = 'your_supabase_key'
-
-    # keys = load_api_keys_from_db()
-    # print("Loaded keys:", keys)
-
-    # # Simulate using cached keys
-    # keys = load_api_keys_from_db()
-    # print("Loaded keys (from cache):", keys)
-    pass

teams/ai_quant/tools/get_current_time.py

@@ -1,28 +0,0 @@
-import datetime
-from zoneinfo import ZoneInfo
-def get_current_time(city: str) -> dict:
-    """Returns the current time in a specified city.
-
-    Args:
-        city (str): The name of the city for which to retrieve the current time.
-
-    Returns:
-        dict: status and result or error msg.
-    """
-
-    if city.lower() == "new york":
-        tz_identifier = "America/New_York"
-    else:
-        return {
-            "status": "error",
-            "error_message": (
-                f"Sorry, I don't have timezone information for {city}."
-            ),
-        }
-
-    tz = ZoneInfo(tz_identifier)
-    now = datetime.datetime.now(tz)
-    report = (
-        f'The current time in {city} is {now.strftime("%Y-%m-%d %H:%M:%S %Z%z")}'
-    )
-    return {"status": "success", "report": report}

teams/ai_quant/tools/get_weather.py

@@ -1,22 +0,0 @@
-def get_weather(city: str) -> dict:
-    """Retrieves the current weather report for a specified city.
-
-    Args:
-        city (str): The name of the city for which to retrieve the weather report.
-
-    Returns:
-        dict: status and result or error msg.
-    """
-    if city.lower() == "new york":
-        return {
-            "status": "success",
-            "report": (
-                "The weather in New York is sunny with a temperature of 25 degrees"
-                " Celsius (77 degrees Fahrenheit)."
-            ),
-        }
-    else:
-        return {
-            "status": "error",
-            "error_message": f"Weather information for '{city}' is not available.",
-        }