feat:查询可用会议室接口更改

yj_room_agent/graph/room_meeting_graph_agent.py

@@ -10,7 +10,7 @@ from yj_room_agent.tools import json_tools, getinfo, params_filter
 from langgraph.graph import StateGraph, END, START
 from langgraph.checkpoint.memory import MemorySaver
 from yj_room_agent.temp_history import msg_history
-
+DEFAULT_QUERY_SIZE = config('DEFAULT_QUERY_SIZE', cast=int, default=30)
 room_llm = ChatOpenAI(
     model=config('MODEL_NAME', default=''),
     base_url=config('MODEL_BASE_URL', default=''),
@@ -119,9 +119,13 @@ def pre_book_meeting_node_handler(state: RoomChatAgentState) -> str:
 def query_avliable_meeting_room_node(state: RoomChatAgentState) -> dict:
     param = state.get('params', {})
     pre_data = state.get('pre_book_meeting_data', {})
-
-    content = getinfo.query_meetingroom(param)
-    result = params_filter.filter_meetingroom(content['data'])
+    param['startTimeStr'] = pre_data.get('start_time', '')
+    param['endTimeStr'] = pre_data.get('end_time', '')
+    param['pageNum'] = 1
+    param['pageSize'] = DEFAULT_QUERY_SIZE
+    content = getinfo.query_avali_meetinroom(param)
+    list = content['data'].get('list', [])
+    result = params_filter.filter_meetingroom(list)
     logger.info(" filter data: {0}".format(result))
     result = params_filter.filter_params(result, "query_meeting_room")
 

yj_room_agent/tools/getinfo.py

@@ -239,6 +239,8 @@ def query_avali_meetinroom(params: dict):
     logger.info("request url => {0}".format(request_url))
     if not params.get("pageSize", None):
         params['pageSize'] = page_size
+    acc_token=get_access_token(tenant_id)
+    params['access_token'] = acc_token
     logger.debug("query_avali_meetingroom params => {0}".format(params))
     respones = requests.get(url=request_url, params=params,
                             headers=header, verify=False)