update

__init__.py

@@ -0,0 +1 @@
+print('\n\n\n\n\n\n\n\n\n&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&')

ai_quant_agent.txt

@@ -0,0 +1,30 @@
+{'after_agent_callback': None,
+ 'after_model_callback': None,
+ 'after_tool_callback': None,
+ 'before_agent_callback': None,
+ 'before_model_callback': <function my_before_model_callback at 0x10e18ba60>,
+ 'before_tool_callback': None,
+ 'code_executor': None,
+ 'description': 'Agent to answer questions about the time and weather in a '
+                'city.',
+ 'disallow_transfer_to_parent': False,
+ 'disallow_transfer_to_peers': False,
+ 'examples': None,
+ 'generate_content_config': None,
+ 'global_instruction': '',
+ 'include_contents': 'default',
+ 'input_schema': None,
+ 'instruction': 'You are a helpful agent who can answer user questions about '
+                'the time and weather in a city.\n'
+                "Before you answer, you should say your name: 'I am "
+                "weather_time_agent'",
+ 'model': 'gemini-2.5-flash-preview-04-17',
+ 'name': 'weather_time_agent',
+ 'output_key': None,
+ 'output_schema': None,
+ 'parent_agent': None,
+ 'planner': None,
+ 'sub_agents': [LlmAgent(name='greeting_agent', description='Handles simple greetings and hellos', parent_agent=None, sub_agents=[], before_agent_callback=None, after_agent_callback=None, 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. \nDo not engage in any other conversation or tasks.\nBefore you answer, you should say your name: 'I am greeting_agent'", global_instruction='', tools=[], generate_content_config=None, disallow_transfer_to_parent=False, disallow_transfer_to_peers=False, include_contents='default', input_schema=None, output_schema=None, output_key=None, planner=None, code_executor=None, examples=None, before_model_callback=None, after_model_callback=None, before_tool_callback=None, after_tool_callback=None),
+                LlmAgent(name='farewell_agent', description='Handles simple farewells and goodbyes', parent_agent=None, sub_agents=[], before_agent_callback=None, after_agent_callback=None, model='gemini-2.5-flash-preview-04-17', instruction="You are the Farewell Agent. Your ONLY task is to provide a polite goodbye message.\nIf not about goodbye, just transfer to weather_time_agent.\nDo not perform any other actions.\nBefore you answer, you should say your name: 'I am farewell_agent'", global_instruction='', tools=[], generate_content_config=None, disallow_transfer_to_parent=False, disallow_transfer_to_peers=False, include_contents='default', input_schema=None, output_schema=None, output_key=None, planner=None, code_executor=None, examples=None, before_model_callback=None, after_model_callback=None, before_tool_callback=None, after_tool_callback=None)],
+ 'tools': [<function get_weather at 0x10e35b380>,
+           <function get_current_time at 0x10e18bce0>]}

{teams → bak}/get_weather/__init__.py

@@ -1,2 +1 @@
 from . import agent 
-from . import load_tools

{teams → bak}/get_weather/agent.py

@@ -1,52 +1,40 @@
 ### adk web ./teams
-
-# load tools from teams_tools，tools 的保存方法 tools['name']=tool
-from get_weather.load_tools import load_tools_from_db
-from get_weather.load_agents import load_agents_from_db
-from get_weather.cache import get_api_key # Import get_api_key
 from google.adk.agents import Agent
-from google.adk.models.lite_llm import LiteLlm
+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
 
-tools = load_tools_from_db()
-
-# def my_before_model_callback(agent, model_input, callback_context, **kwargs):
 def my_before_model_callback(callback_context, **kwargs):
     os.environ["GOOGLE_API_KEY"] = get_api_key()  # 直接设置环境变量
-
-# load not_root_agent from teams_agents
-agents = load_agents_from_db()
-for i in range(len(agents)):
-    agents[i].before_model_callback = my_before_model_callback
+print('888888888888888888888888888888888888888888888')
 
 greeting_agent = Agent(
     name="greeting_agent",
-    model="gemini-2.0-flash-exp",
+    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'",
-    # Crucial for delegation: Clear description of capability
     description="Handles simple greetings and hellos",
-    before_model_callback=my_before_model_callback # Add the callback here
- )
+    before_model_callback=my_before_model_callback
+)
 
 farewell_agent = Agent(
-   name="farewell_agent",
-   model="gemini-2.0-flash-exp",
-    
+    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'",
-    # Crucial for delegation: Clear description of capability
+                "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",
-    before_model_callback=my_before_model_callback # Add the callback here
 )
 
-# load root_agent from teams_agents
-root_agent = Agent(
+print("\n\n\n\n\n\n\n\n&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&\nroot_agent init")
+root_agent=Agent(
     name="weather_time_agent",
-    model="gemini-2.0-flash-exp",
+    model="gemini-2.5-flash-preview-04-17",
     description=(
         "Agent to answer questions about the time and weather in a city."
     ),
@@ -54,8 +42,12 @@ root_agent = Agent(
         "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'"
     ),
-    tools=[tools['get_weather'],tools['get_current_time']],
-    # sub_agents=[greeting_agent, farewell_agent],
-    sub_agents=agents,
-    before_model_callback=my_before_model_callback # Add the callback here
+    # 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/team.py

@@ -0,0 +1,33 @@
+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

get_weather_agent.txt

@@ -0,0 +1,29 @@
+{'after_agent_callback': None,
+ 'after_model_callback': None,
+ 'after_tool_callback': None,
+ 'before_agent_callback': None,
+ 'before_model_callback': <function my_before_model_callback at 0x1169d5e40>,
+ 'before_tool_callback': None,
+ 'code_executor': None,
+ 'description': 'Agent to answer questions about the time and weather in a '
+                'city.',
+ 'disallow_transfer_to_parent': False,
+ 'disallow_transfer_to_peers': False,
+ 'examples': None,
+ 'generate_content_config': None,
+ 'global_instruction': '',
+ 'include_contents': 'default',
+ 'input_schema': None,
+ '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'",
+ 'model': 'gemini-2.5-flash-preview-04-17',
+ 'name': 'weather_time_agent',
+ 'output_key': None,
+ 'output_schema': None,
+ 'parent_agent': None,
+ 'planner': None,
+ 'sub_agents': [LlmAgent(name='greeting_agent', description='Handles simple greetings and hellos', parent_agent=LlmAgent(name='weather_time_agent', description='Agent to answer questions about the time and weather in a city.', parent_agent=None, sub_agents=[LlmAgent(name='greeting_agent', description='Handles simple greetings and hellos', parent_agent=LlmAgent(name='weather_time_agent', description='Agent to answer questions about the time and weather in a city.', parent_agent=None, sub_agents=[...], before_agent_callback=None, after_agent_callback=None, model='gemini-2.5-flash-preview-04-17', 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=[<function get_weather at 0x116afd300>, <function get_current_time at 0x116afd080>], generate_content_config=None, disallow_transfer_to_parent=False, disallow_transfer_to_peers=False, include_contents='default', input_schema=None, output_schema=None, output_key=None, planner=None, code_executor=None, examples=None, before_model_callback=<function my_before_model_callback at 0x1169d5e40>, after_model_callback=None, before_tool_callback=None, after_tool_callback=None), sub_agents=[], before_agent_callback=None, after_agent_callback=None, 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'", global_instruction='', tools=[], generate_content_config=None, disallow_transfer_to_parent=False, disallow_transfer_to_peers=False, include_contents='default', input_schema=None, output_schema=None, output_key=None, planner=None, code_executor=None, examples=None, before_model_callback=<function my_before_model_callback at 0x1169d5e40>, after_model_callback=None, before_tool_callback=None, after_tool_callback=None), LlmAgent(name='farewell_agent', description='Handles simple farewells and goodbyes', parent_agent=LlmAgent(name='weather_time_agent', description='Agent to answer questions about the time and weather in a city.', parent_agent=None, sub_agents=[...], before_agent_callback=None, after_agent_callback=None, model='gemini-2.5-flash-preview-04-17', 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=[<function get_weather at 0x116afd300>, <function get_current_time at 0x116afd080>], generate_content_config=None, disallow_transfer_to_parent=False, disallow_transfer_to_peers=False, include_contents='default', input_schema=None, output_schema=None, output_key=None, planner=None, code_executor=None, examples=None, before_model_callback=<function my_before_model_callback at 0x1169d5e40>, after_model_callback=None, before_tool_callback=None, after_tool_callback=None), sub_agents=[], before_agent_callback=None, after_agent_callback=None, 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_agentDo not perform any other actions.before you answer, you should say your name: 'I am farewell_agent'", global_instruction='', tools=[], generate_content_config=None, disallow_transfer_to_parent=False, disallow_transfer_to_peers=False, include_contents='default', input_schema=None, output_schema=None, output_key=None, planner=None, code_executor=None, examples=None, before_model_callback=<function my_before_model_callback at 0x1169d5e40>, after_model_callback=None, before_tool_callback=None, after_tool_callback=None)], before_agent_callback=None, after_agent_callback=None, model='gemini-2.5-flash-preview-04-17', 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=[<function get_weather at 0x116afd300>, <function get_current_time at 0x116afd080>], generate_content_config=None, disallow_transfer_to_parent=False, disallow_transfer_to_peers=False, include_contents='default', input_schema=None, output_schema=None, output_key=None, planner=None, code_executor=None, examples=None, before_model_callback=<function my_before_model_callback at 0x1169d5e40>, after_model_callback=None, before_tool_callback=None, after_tool_callback=None), sub_agents=[], before_agent_callback=None, after_agent_callback=None, 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'", global_instruction='', tools=[], generate_content_config=None, disallow_transfer_to_parent=False, disallow_transfer_to_peers=False, include_contents='default', input_schema=None, output_schema=None, output_key=None, planner=None, code_executor=None, examples=None, before_model_callback=<function my_before_model_callback at 0x1169d5e40>, after_model_callback=None, before_tool_callback=None, after_tool_callback=None),
+                LlmAgent(name='farewell_agent', description='Handles simple farewells and goodbyes', parent_agent=LlmAgent(name='weather_time_agent', description='Agent to answer questions about the time and weather in a city.', parent_agent=None, sub_agents=[LlmAgent(name='greeting_agent', description='Handles simple greetings and hellos', parent_agent=LlmAgent(name='weather_time_agent', description='Agent to answer questions about the time and weather in a city.', parent_agent=None, sub_agents=[...], before_agent_callback=None, after_agent_callback=None, model='gemini-2.5-flash-preview-04-17', 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=[<function get_weather at 0x116afd300>, <function get_current_time at 0x116afd080>], generate_content_config=None, disallow_transfer_to_parent=False, disallow_transfer_to_peers=False, include_contents='default', input_schema=None, output_schema=None, output_key=None, planner=None, code_executor=None, examples=None, before_model_callback=<function my_before_model_callback at 0x1169d5e40>, after_model_callback=None, before_tool_callback=None, after_tool_callback=None), sub_agents=[], before_agent_callback=None, after_agent_callback=None, 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'", global_instruction='', tools=[], generate_content_config=None, disallow_transfer_to_parent=False, disallow_transfer_to_peers=False, include_contents='default', input_schema=None, output_schema=None, output_key=None, planner=None, code_executor=None, examples=None, before_model_callback=<function my_before_model_callback at 0x1169d5e40>, after_model_callback=None, before_tool_callback=None, after_tool_callback=None), LlmAgent(name='farewell_agent', description='Handles simple farewells and goodbyes', parent_agent=LlmAgent(name='weather_time_agent', description='Agent to answer questions about the time and weather in a city.', parent_agent=None, sub_agents=[...], before_agent_callback=None, after_agent_callback=None, model='gemini-2.5-flash-preview-04-17', 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=[<function get_weather at 0x116afd300>, <function get_current_time at 0x116afd080>], generate_content_config=None, disallow_transfer_to_parent=False, disallow_transfer_to_peers=False, include_contents='default', input_schema=None, output_schema=None, output_key=None, planner=None, code_executor=None, examples=None, before_model_callback=<function my_before_model_callback at 0x1169d5e40>, after_model_callback=None, before_tool_callback=None, after_tool_callback=None), sub_agents=[], before_agent_callback=None, after_agent_callback=None, 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_agentDo not perform any other actions.before you answer, you should say your name: 'I am farewell_agent'", global_instruction='', tools=[], generate_content_config=None, disallow_transfer_to_parent=False, disallow_transfer_to_peers=False, include_contents='default', input_schema=None, output_schema=None, output_key=None, planner=None, code_executor=None, examples=None, before_model_callback=<function my_before_model_callback at 0x1169d5e40>, after_model_callback=None, before_tool_callback=None, after_tool_callback=None)], before_agent_callback=None, after_agent_callback=None, model='gemini-2.5-flash-preview-04-17', 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=[<function get_weather at 0x116afd300>, <function get_current_time at 0x116afd080>], generate_content_config=None, disallow_transfer_to_parent=False, disallow_transfer_to_peers=False, include_contents='default', input_schema=None, output_schema=None, output_key=None, planner=None, code_executor=None, examples=None, before_model_callback=<function my_before_model_callback at 0x1169d5e40>, after_model_callback=None, before_tool_callback=None, after_tool_callback=None), sub_agents=[], before_agent_callback=None, after_agent_callback=None, 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_agentDo not perform any other actions.before you answer, you should say your name: 'I am farewell_agent'", global_instruction='', tools=[], generate_content_config=None, disallow_transfer_to_parent=False, disallow_transfer_to_peers=False, include_contents='default', input_schema=None, output_schema=None, output_key=None, planner=None, code_executor=None, examples=None, before_model_callback=<function my_before_model_callback at 0x1169d5e40>, after_model_callback=None, before_tool_callback=None, after_tool_callback=None)],
+ 'tools': [<function get_weather at 0x116afd300>,
+           <function get_current_time at 0x116afd080>]}

team_framework/__init__.py

@@ -0,0 +1 @@
+# from . import load_team_config

team_framework/load_agents.py

@@ -0,0 +1,32 @@
+from google.adk.agents import Agent
+import os
+
+def my_before_model_callback(callback_context, **kwargs):
+    os.environ["GOOGLE_API_KEY"]='AIzaSyBYPwMLibOIoM-SQHuWjZ-V4LLmmMxqhm8'
+#     os.environ["GOOGLE_API_KEY"] = get_api_key()  # 直接设置环境变量
+
+def load_agent(item):
+    print(f"\nload_agent item: {item['name']}", item)
+    _agent = Agent(
+        name = item['name'],
+        model = item['model'],
+        instruction = item['instruction'],
+        description = item['description'],
+        before_model_callback=my_before_model_callback
+    )
+    return _agent
+
+def load_agents(agents_dict:dict):
+    print("\n\n\n######## load_agents\nagents_dict: ", agents_dict)
+    ret_agents = {}
+    for item  in agents_dict:
+        ret_agents[item['name']] = load_agent(item)
+        # ret_agents.append(load_agent(item))
+    print("\n\n\n@@@@@@@@@@@@@@@@@@@@@\nload_agents\nret_agents.greeting_agent: ", ret_agents['greeting_agent'])
+    print("\n\n\n@@@@@@@@@@@@@@@@@@@@@\nload_agents\nret_agents[farewell_agent]: ", ret_agents['farewell_agent'])
+    return ret_agents
+
+def load_root_agent(_root_agent):
+    ret_agent = load_agent(_root_agent)
+    print("\n\n\n######## load_root_agent\nret_agent: ", ret_agent)
+    return ret_agent

team_framework/load_team_config.py

@@ -0,0 +1,93 @@
+from .supabase_cli import get_supabase_client
+from supabase import Client
+from google.adk.agents import Agent
+
+def load_agent_config(db_item):
+    # print('\nload_agent_config...')
+    # print('\n\n\n\n\n***************************\ndb_item',db_item)
+    agent_config={
+        "name": db_item['name'],
+        "model": db_item['model'],
+        "instruction": db_item['instruction'],
+        "tools": db_item['tools'],
+        "sub_agents": db_item['sub_agents'],
+        "description": db_item['description']
+    }
+    tools_config=[]
+    if db_item['tools']:
+        agent_config['tools'] = db_item['tools']
+        tools_config = list(set(tools_config + db_item['tools']))
+    return { 'agent_config': agent_config, 'tools_config': tools_config}
+
+def creat_agent() -> Agent:
+    pass
+def load_team_config():
+    a_team_config = {
+        "agents":[],
+        "root_agent": None,
+        "tools":[]
+    }
+
+    print("Loading team_config from database...")
+    supabase: Client = get_supabase_client()
+    if not supabase:
+        return {}
+    try:
+        response = supabase.from_("team_get_weather").select("*").order("sort_id").execute()
+        for item in response.data:
+            if item['is_root']:
+                _agent_config = load_agent_config(item)
+                print('_agent_config',_agent_config)
+                a_team_config['root_agent'] = _agent_config['agent_config']
+                a_team_config['tools'] = list(set(a_team_config['tools'] + _agent_config['tools_config']))
+            else:
+                _agent_config = load_agent_config(item)
+                a_team_config['agents'].append(_agent_config['agent_config'])
+                a_team_config['tools'] = list(set(a_team_config['tools'] + _agent_config['tools_config']))
+        # print('\n\na_team_config',a_team_config)
+    except Exception as e:
+        print(f"Error loading API keys from database: {e}")
+        a_team_config = {} # Clear cache on error
+    return a_team_config
+
+"""
+a_team_config
+{
+    'agents': [
+        {
+            '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.\nIf not about goodbye, just transfer to weather_time_agent.\nDo not perform any other actions.\nBefore you answer, you should say your name: 'I am farewell_agent'", 
+            'tools': None, 
+            'sub_agents': None, 
+            'description': 'Handles simple farewells and goodbyes'
+        }, 
+        {
+            '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. \nDo not engage in any other conversation or tasks.\nBefore you answer, you should say your name: 'I am greeting_agent'", 
+            'tools': None, 
+            'sub_agents': None, 
+            'description': 'Handles simple greetings and hellos'
+        }
+    ], 
+    'root_agent': {
+        'name': 'weather_time_agent', 
+        'model': 'gemini-2.5-flash-preview-04-17', '
+        instruction': "You are a helpful agent who can answer user questions about the time and weather in a city.\nBefore you answer, you should say your name: 'I am weather_time_agent'", 
+        'tools': [
+            'get_weather', 
+            'get_current_time'
+        ], 
+        'sub_agents': [
+            'greeting_agent', 
+            'farewell_agent'
+        ], 
+        'description': 'Agent to answer questions about the time and weather in a city.'
+    }, 
+    'tools': [
+        'get_weather', 
+        'get_current_time'
+    ]    
+}
+"""

teams/get_weather/load_agents.py → team_framework/load_tools.py

@@ -1,13 +1,12 @@
-import importlib, os
-from supabase import create_client, Client
-from google.adk.agents import Agent
+import importlib
 
-def load_agents_from_array(arr:list):
+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}'
+        # parent_module_path = '.'.join(__name__.split('.')[:-1])
+        # module_path = f'{parent_module_path}.tools.{item}'
+        module_path = f'teams.get_weather.tools.{item}'
         try:
             module = importlib.import_module(module_path)
             # 函数名预期与工具名称相同
@@ -20,40 +19,12 @@ def load_agents_from_array(arr:list):
             print(f"在模块 {module_path} 中找不到函数 '{item}': {e}")
     return ret
 
-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_tools_from_db():
+#     tool_arr = ["get_weather", "get_current_time"]
+#     return load_tools_from_array(tool_arr)
 
-def load_agents_from_db():
-    """Loads agents from the database."""
-    supabase: Client = get_supabase_client()
-    if not supabase:
-        return {}
-
-    try:
-        response = supabase.from_("team_get_weather").select("*").eq("is_root", False).execute()
-        if response.data and response.data[0]:
-            ret = []
-            for item in response.data:
-                agent = Agent(
-                    name=item['name'], 
-                    model=item['model'], 
-                    instruction=item['instruction'], 
-                    # tools=item['tools'], 
-                    description=item['description']
-                )
-                ret.append(agent)
-            return ret
-        else:
-            print("Unexpected database response structure or no data found.")
-            ret = {}
-
-    except Exception as e:
-        print(f"Error loading agents from database: {e}")
-        ret = {} # Clear cache on error
-    return ret
+def load_tools(tools_dict:dict):
+    print("\n\n\n######## load_tools\ntools_dict: ", tools_dict)
+    # load_tools_from_array(tools_dict)
+    return load_tools_from_array(tools_dict)
+    return tools_dict

team_framework/supabase_cli.py

@@ -0,0 +1,11 @@
+from supabase import create_client, Client
+import os
+
+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)

teams/ai_quant/__init__.py

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

teams/ai_quant/agent.py

@@ -0,0 +1,148 @@
+import os
+from google.adk.agents import Agent
+from supabase import create_client, Client
+from .tools.get_current_time import get_current_time
+from .tools.get_weather import get_weather
+def my_before_model_callback(callback_context, **kwargs):
+    # os.environ["GOOGLE_API_KEY"] = get_api_key()  # 直接设置环境变量
+    os.environ["GOOGLE_API_KEY"] = "AIzaSyDCbN6WDNIEtB8yJmNX40ScechQu6mCZoo"  # 直接设置环境变量
+
+print('99999999999999999999999999999999999999999')
+
+# 从环境变量获取 Supabase 配置
+# 建议将敏感信息通过环境变量传递，而不是直接硬编码或从文件中读取
+# 这里为了演示，直接使用了从 .env 文件读取到的值
+# 在实际生产环境中，请确保通过安全的方式加载配置
+SUPABASE_URL = os.environ.get("SUPABASE_URL", "https://bbvyxjvleoizsipxydsm.supabase.co")
+SUPABASE_KEY = os.environ.get("SUPABASE_KEY", "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6ImJidnl4anZsZW9изsipxydsmIiwicm9sZSI6ImFub24iLCJpYXQiOjE3NDQ1MzkwNjIsImV4cCI6MjA2MDExNTA2Mn0.cbUNhiCHXqIaW4leU_vGTKVI51gLVvsFOVWYS_vJrA4")
+DB_TABLE_NAME = os.environ.get("DB_TABLE_NAME", "team_quant")
+
+# 初始化 Supabase 客户端
+supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
+
+def get_all_agents_from_db() -> dict:
+    """
+    从 Supabase 数据库中提取所有 agent 记录，并形成字典。
+    """
+    try:
+        # 从指定的表中查询所有数据
+        response = supabase.table(DB_TABLE_NAME).select("*").order('sort_id', desc=False).execute()
+        print('\n\n\n\n\nresponse:\n', response)
+
+        # 检查响应是否成功
+        if response.data:
+            # 将查询结果转换为字典，这里假设每条记录都有一个唯一的 'id' 或其他键
+            # 如果没有合适的唯一键，可能需要调整字典的构建方式
+            agents_dict = {record.get("id"): record for record in response.data}
+            print(f"成功从数据库提取 {len(response.data)} 条 agent 记录。")
+            return agents_dict
+        else:
+            print("从数据库提取 agent 记录失败或没有记录。")
+            return {}
+    except Exception as e:
+        print(f"提取 agent 记录时发生错误: {e}")
+        return {}
+
+# 示例用法 (可选，根据需要决定是否保留)
+# if __name__ == "__main__":
+#     all_agents = get_all_agents_from_db()
+#     print("提取到的 agent 字典:")
+#     print(all_agents)
+
+# 从数据库加载所有 agents 数据 (字典形式)
+all_agents_data_dict = get_all_agents_from_db()
+print("\n\n\n\n\n\n\n从数据库加载的 agents 数据 all_agents_data_dict:\n",all_agents_data_dict)
+
+# 将字典值转换为列表并按 sort_id 排序
+agent_records = sorted(all_agents_data_dict.values(), key=lambda x: x.get("sort_id", float('inf')))
+
+# 初始化 root_ageint_idx 变量
+root_ageint_idx = -1
+
+# 查找 root agent 的索引
+if agent_records:
+    for index, agent_data in enumerate(agent_records):
+        if agent_data.get("is_root", False): # 假设数据库记录包含 is_root 字段
+            root_ageint_idx = index
+            print(f"找到 root agent 的索引: {root_ageint_idx}")
+            break # 假设只有一个 root agent
+
+# 创建 Agent 对象并存储到按名称索引的字典中
+agents_by_name = {}
+loaded_agents_list = [] # Keep a list for the final output if needed
+if agent_records:
+    for agent_data in agent_records:
+        try:
+            # 假设数据库记录包含 name, model, instruction, description, sub_agents, sort_id, is_root 字段
+            agent_name = agent_data.get("name", f"agent_{agent_data.get('id', 'unknown')}")
+            agent = Agent(
+                name=agent_name,
+                model=agent_data.get("model", "gemini-2.0-flash-exp"),
+                instruction=agent_data.get("instruction", ""),
+                description=agent_data.get("description", ""),
+                # sub_agents will be added in the next step
+                # callbacks etc. can be added here if present in data
+            )
+            # Map tool names from database to actual tool functions
+            tool_map = {"get_weather": get_weather, "get_current_time": get_current_time}
+            tools_list = [tool_map.get(tool_name) for tool_name in agent_data.get("tools") or [] if tool_name in tool_map]
+            agent.tools = tools_list # Use the mapped tool functions
+
+            agents_by_name[agent_name] = agent
+            loaded_agents_list.append(agent) # Add to list if needed later
+            print(f"成功创建 agent 对象: {agent.name}")
+        except Exception as e:
+            print(f"创建 agent 对象时发生错误 (数据: {agent_data}): {e}")
+
+# 处理 sub_agents 引用
+if agent_records:
+    for agent_data in agent_records:
+        agent_name = agent_data.get("name", f"agent_{agent_data.get('id', 'unknown')}")
+        current_agent = agents_by_name.get(agent_name)
+        if current_agent and "sub_agents" in agent_data and isinstance(agent_data["sub_agents"], list):
+            current_agent.sub_agents = []
+            for sub_agent_name in agent_data["sub_agents"]:
+                sub_agent = agents_by_name.get(sub_agent_name)
+                if sub_agent:
+                    current_agent.sub_agents.append(sub_agent)
+                    print(f"为 agent '{current_agent.name}' 添加 sub-agent: '{sub_agent.name}'")
+                else:
+                    print(f"警告: 未找到 sub-agent '{sub_agent_name}' (被 agent '{current_agent.name}' 引用)")
+
+print(f"总共加载并关联了 {len(agents_by_name)} 个 agents。")
+
+# 您现在可以使用 agents_by_name 字典或 loaded_agents_list 列表中的 Agent 对象
+# 例如，如果您需要一个列表：
+loaded_agents = loaded_agents_list
+
+# 根据找到的索引设置 root_agent
+root_agent = None
+if 0 <= root_ageint_idx < len(loaded_agents_list):
+    root_agent = loaded_agents_list[root_ageint_idx]
+    # root_agent.before_model_callback=my_before_model_callback
+    print(f"\n\n\n\n\n已设置 root_agent 为: {root_agent.name}")
+else:
+    print("警告: 未找到 root agent 或索引无效，使用默认 root_agent。")
+    # 如果没有找到 root agent，可以考虑使用一个默认的 root_agent 或者抛出错误
+    root_agent = Agent(
+        name="default_root_agent",
+        model="gemini-2.0-flash-exp",
+        instruction="You are a helpful assistant.",
+        description="You are a helpful assistant.",
+        # before_model_callback=my_before_model_callback,
+        tools=[]
+    )
+
+# 轮询 loaded_agents_list
+for agent in loaded_agents_list:
+    agent.before_model_callback=my_before_model_callback
+    if agent.sub_agents:
+        for sub_agent in agent.sub_agents:
+            sub_agent.parent_agent = agent
+
+import pprint
+print('\n\n\n\n*************************************\n\nroot_agent dict: \n')
+pprint.pprint(root_agent.__dict__) # 尝试打印对象的字典表示
+
+
+

teams/ai_quant/readme.md

@@ -0,0 +1,6 @@
+- 我要搭建一个google agent development kit 项目
+- 从supabase数据库里的数据表内容，加载agent的配置信息，生成相应的agent实例
+- supabase 数据库配置信息、团队基本信息存储在 .env 文件里
+- 注意1：is_root字段为True的agent，是团队的根agent——root_agent，其他agent为可选子agents，root_agent必须有
+- 注意2：数据库表里：tools字段（jsonb）里["google_search"]表示agent可以使用google_search这个tool；数据库表里：subbase_table字段（jsonb）里["searcher"]表示agent有子agent，名为searcher
+

teams/ai_quant/team_quant_rows.sql

@@ -0,0 +1,3 @@
+INSERT INTO "public"."team_quant" ("id", "model", "instruction", "tools", "description", "created_at", "name", "is_root", "sub_agents", "sort_id", "title", "global_instruction") VALUES ('bba6efef-9db5-44c4-817b-df69f1b3a453', 'gemini-2.0-pro-exp', '你是谷歌搜索专家，专门搜索和金融量化交易相关的信息。
+仅执行搜索操作，不处理其他任务', '["searcher"]', '你是搜索专家，专门搜索和金融量化交易相关的信息', '2025-05-11 12:37:23.847852+00', 'searcher', 'false', null, '1', '金融量化信息检索专家', null), ('c78579d7-6fe5-4c73-860b-777ac202fc28', 'gemini-2.0-pro-exp', '你是量化交易团队的调度员。
+将搜索任务委派给google_searcher', null, '量化交易团队的调度员', '2025-05-10 09:58:58.341622+00', 'dispatcher', 'true', '["searcher"]', null, '调度员', '量化交易团队成员');

teams/ai_quant/tools/get_current_time.py

@@ -0,0 +1,28 @@
+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

@@ -0,0 +1,22 @@
+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.",
+        }

test.py

@@ -0,0 +1,5 @@
+d=1
+a={"b":{"c":d}}
+print(a["b"]["c"])
+d=2
+print(a["b"]["c"])