Skip to content

Commit dc51b33

Browse files
authored
Removed unnecessarily duplicated props for agent context (#41)
* removed unnecessarily duplicated props * imports
1 parent 09d8cac commit dc51b33

File tree

6 files changed

+1
-20
lines changed

6 files changed

+1
-20
lines changed

examples/cat-lounge/backend/app/cat_agent.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@
1717

1818
from .cat_state import CatState
1919
from .cat_store import CatStore
20-
from .memory_store import MemoryStore
2120
from .widgets.name_suggestions_widget import CatNameSuggestion, build_name_suggestions_widget
2221
from .widgets.profile_card_widget import build_profile_card_widget, profile_widget_copy_text
2322

@@ -72,9 +71,7 @@
7271

7372
class CatAgentContext(AgentContext):
7473
model_config = ConfigDict(arbitrary_types_allowed=True)
75-
store: Annotated[MemoryStore, Field(exclude=True)]
7674
cats: Annotated[CatStore, Field(exclude=True)]
77-
request_context: dict[str, Any]
7875

7976

8077
async def _get_state(ctx: RunContextWrapper[CatAgentContext]) -> CatState:

examples/metro-map/backend/app/agents/metro_map_agent.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,9 +76,7 @@
7676

7777
class MetroAgentContext(AgentContext):
7878
model_config = ConfigDict(arbitrary_types_allowed=True)
79-
store: Annotated[MemoryStore, Field(exclude=True)]
8079
metro: Annotated[MetroMapStore, Field(exclude=True)]
81-
request_context: Annotated[RequestContext, Field(exclude=True)]
8280

8381

8482
class MapResult(BaseModel):

examples/metro-map/backend/app/server.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@
1818
AssistantMessageItem,
1919
Attachment,
2020
ClientEffectEvent,
21-
ClientToolCallItem,
2221
HiddenContextItem,
2322
ThreadItemDoneEvent,
2423
ThreadItemReplacedEvent,

examples/news-guide/backend/app/agents/event_finder_agent.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,9 +49,7 @@
4949

5050
class EventFinderContext(AgentContext):
5151
model_config = ConfigDict(arbitrary_types_allowed=True)
52-
store: Annotated[MemoryStore, Field(exclude=True)]
5352
events: Annotated[EventStore, Field(exclude=True)]
54-
request_context: Annotated[RequestContext, Field(exclude=True, default_factory=RequestContext)]
5553

5654

5755
class EventKeywords(BaseModel):

examples/news-guide/backend/app/agents/news_agent.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,6 @@
1717
from ..agents.event_finder_agent import event_finder_agent
1818
from ..agents.puzzle_agent import puzzle_agent
1919
from ..data.article_store import ArticleMetadata, ArticleRecord, ArticleStore
20-
from ..memory_store import MemoryStore
21-
from ..request_context import RequestContext
2220
from ..widgets.article_list_widget import build_article_list_widget
2321

2422
logging.basicConfig(level=logging.INFO)
@@ -84,9 +82,7 @@
8482

8583
class NewsAgentContext(AgentContext):
8684
model_config = ConfigDict(arbitrary_types_allowed=True)
87-
store: Annotated[MemoryStore, Field(exclude=True)]
8885
articles: Annotated[ArticleStore, Field(exclude=True)]
89-
request_context: Annotated[RequestContext, Field(exclude=True)]
9086

9187

9288
# -- Structured results for tool calls --------------------------------------

examples/news-guide/backend/app/agents/puzzle_agent.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,8 @@
11
from __future__ import annotations
22

3-
from typing import Annotated
4-
53
from agents import Agent
64
from chatkit.agents import AgentContext
7-
from pydantic import ConfigDict, Field
8-
9-
from ..memory_store import MemoryStore
10-
from ..request_context import RequestContext
5+
from pydantic import ConfigDict
116

127
INSTRUCTIONS = """
138
You host Foxhollow's Coffee Break Puzzle Corner — a cheerful diversion for readers steeped
@@ -47,8 +42,6 @@
4742

4843
class PuzzleAgentContext(AgentContext):
4944
model_config = ConfigDict(arbitrary_types_allowed=True)
50-
store: Annotated[MemoryStore, Field(exclude=True)]
51-
request_context: Annotated[RequestContext, Field(exclude=True, default_factory=RequestContext)]
5245

5346

5447
puzzle_agent = Agent[PuzzleAgentContext](

0 commit comments

Comments
 (0)