Skip to content

Commit 786efa1

Browse files
Merge pull request #11 from iChintanSoni/main
fix: update beeai-sdk version to 0.3.7 in uv.lock and refactor import…
2 parents 1889810 + 4456b40 commit 786efa1

1 file changed

Lines changed: 4 additions & 8 deletions

File tree

src/beeai_agents/agent.py

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@
66

77
from beeai_framework.adapters.openai import OpenAIChatModel
88
from beeai_framework.backend.types import ChatModelParameters
9-
from beeai_framework.agents.experimental import RequirementAgent
10-
from beeai_framework.agents.experimental.requirements.conditional import ConditionalRequirement
9+
from beeai_framework.agents.requirement import RequirementAgent
10+
from beeai_framework.agents.requirement.requirements.conditional import ConditionalRequirement
1111
from beeai_framework.agents.requirement.events import RequirementAgentFinalAnswerEvent
1212
from beeai_framework.agents.types import AgentExecutionConfig
1313
from beeai_framework.backend.message import UserMessage, AssistantMessage
@@ -93,7 +93,6 @@ def is_casual(msg: str) -> bool:
9393
detail=AgentDetail(
9494
interaction_mode="multi-turn",
9595
user_greeting="Hi! Try out BeeAI features with me — upload a doc, search the web, or tweak my settings.",
96-
version="0.0.15",
9796
tools=[
9897
AgentDetailTool(
9998
name="Think",
@@ -203,7 +202,6 @@ async def beeai_showcase_agent(
203202
- **Session History:** Stores messages in platform context for persistent conversation history.
204203
"""
205204

206-
# Store incoming message for session history
207205
await context.store(input)
208206

209207
thinking_mode = True
@@ -468,7 +466,6 @@ def handle_final_answer_stream(data: RequirementAgentFinalAnswerEvent, meta: Eve
468466
)
469467
yield citation.citation_metadata(citations=citations)
470468

471-
# Store response in context for session history
472469
response_message = AgentMessage(text=response_text)
473470
await context.store(response_message)
474471

@@ -485,16 +482,15 @@ def handle_final_answer_stream(data: RequirementAgentFinalAnswerEvent, meta: Eve
485482
error_msg = f"Error processing request: {e}"
486483
yield error_msg
487484

488-
# Store error message too
489485
await context.store(AgentMessage(text=error_msg))
490486

491487
def run():
492488
"""Start the server with context storage enabled"""
493489
server.run(
494490
host=os.getenv("HOST", "127.0.0.1"),
495491
port=int(os.getenv("PORT", 8000)),
496-
context_store=PlatformContextStore() # Enable session history
492+
context_store=PlatformContextStore()
497493
)
498494

499495
if __name__ == "__main__":
500-
run()
496+
run()

0 commit comments

Comments
 (0)