Choosing elements from structured task outputs as context to next task

[ajay-bhargava]

Hello,
I would like to limit the task output from a previous task to the next task. I tried to specify the class attributes from the output_pydantic for the Agent but that didn't work

class Website(BaseModel):
	website: str = Field(description="The website URL")
	snippet: str = Field(description="A snippet of the website content")

@CrewBase
class Search():
	"""Search crew"""

	agents_config = 'config/agents.yaml'
	tasks_config = 'config/tasks.yaml'

	ollama_client = LLM(model="ollama/llama3:latest", base_url="http://localhost:11434")

	@agent
	def brevity_agent(self) -> Agent:
		return Agent(
			config=self.agents_config['brevity_agent'],
			llm=self.ollama_client,
			verbose=True
		)
	
	@agent
	def web_researcher(self) -> Agent:
		return Agent(
			config=self.agents_config['web_researcher'],
			tools=[SerperDevTool(
				n_results=3
			)],
			llm=self.ollama_client,
			verbose=True,
			output_pydantic=Website
		)
	
	@agent
	def web_scraper(self) -> Agent:
		return Agent(
			config=self.agents_config['web_scraper'],
			tools=[ScrapeWebsiteTool()],
			llm=self.ollama_client,
			verbose=True,
		)


	@task
	def brevity_task(self) -> Task:
		return Task(
			config=self.tasks_config['brevity_task'],
		)
	
	@task
	def web_researcher_task(self) -> Task:
		return Task(
			config=self.tasks_config['search_task'],
			context=[self.web_researcher.output.website.content]
		)

I'd like to choose specific keys and values to be the context from the previous task to the next task using CrewAI. Any tips?