from crewai.project import CrewBase, agent, crew, task
from langfuse import Langfuse
import openlit
from tools import (x,y)
@CrewBase
class AnalyzerCrew():
 agents_config = 'config/agents.yaml'
 tasks_config = 'config/tasks.yaml'
 @agent
 def planner(self) -> Agent:
 return Agent(
 config=self.agents_config['planner'],
 tools=[
 x,y
 ],
 verbose=True,
 allow_delegation=False,
 max_iter=2,
 max_rpm=15,
 max_execution_time=300
 )
 @task
 def research_planning(self) -> Task:
 return Task(
 config=self.tasks_config['research_planning'],
 agent=self.planner(),
 output_pydantic=ResearchPlan,
 output_file='research_plan.json'
 )
 @crew
 def crew(self) -> Crew:
 """Creates the CTI research planning crew - first stage of multi-agent workflow"""
 return Crew(
 agents=self.agents,
 tasks=self.tasks,
 process=Process.sequential,
 verbose=True,
 planning=True,
 planning_llm="openai/gpt-4o-mini"
 )
_langfuse_client = Langfuse(
 public_key=config.public_key,
 secret_key=config.secret_key,
 host=config.host,
 debug=config.debug
)
openlit.init(
 tracer=client._otel_tracer,
 disable_batch=True
)
with _langfuse_client.start_as_current_span(
 name="Analysis Workflow"
) as span:
 # Initialize crew
 crew = AnalyzerCrew()
 print(f"✅ Crew initialized successfully")
 
 # Run analysis
 result = crew.crew().kickoff(
 inputs={'text': text, 'output_dir': output_dir}
 )