RAG issues with Agents: Need help
hey all, im keen to learn from others on this issue ive been facing. I have the agent deployed in Google Agent Engine, the agent uses RAG as part of its process. Now I have a pipeline where I drop in a file and it will create a RAG corpus for me and do some basic filtering for me, once this is done the agent is called and it has to use the RAG id that was automatically created. Ive tried injecting the RAG ID into the state session but not having much luck, the Agent is using the last hardcoded RAG ID i had before deployment. how can I get the agent to grab the new RAG ID. code for how I create a new session: session = await self.remote_app.async_create_session( user_id=user_id, state=['rag_corpus_id':corpus_id']) and in my rag tool i do something like this: corpus_id_number = tool_context.state.get("rag_corpus_id")