Skip to content

Commit

Permalink
[Feat] Add streaming agents (#277)
Browse files Browse the repository at this point in the history
* format

* fix ReAct

* support streaming agents

* fix type hints

* update
  • Loading branch information
braisedpork1964 authored Nov 25, 2024
1 parent 73c8326 commit 44465ce
Show file tree
Hide file tree
Showing 7 changed files with 402 additions and 337 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,7 @@ print(dumped_memory['memory'])
Clear the memory of this session(`session_id=0` by default):

```python
agent.memory.reset()
agent.reset()
```

### Custom Message Aggregation
Expand Down
32 changes: 28 additions & 4 deletions lagent/agents/__init__.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,33 @@
from .agent import Agent, AgentDict, AgentList, AsyncAgent, AsyncSequential, Sequential
from .agent import (
Agent,
AgentDict,
AgentList,
AsyncAgent,
AsyncSequential,
AsyncStreamingAgent,
AsyncStreamingSequential,
Sequential,
StreamingAgent,
StreamingSequential,
)
from .react import AsyncReAct, ReAct
from .stream import AgentForInternLM, AsyncAgentForInternLM, AsyncMathCoder, MathCoder

__all__ = [
'Agent', 'AgentDict', 'AgentList', 'AsyncAgent', 'AgentForInternLM',
'AsyncAgentForInternLM', 'MathCoder', 'AsyncMathCoder', 'ReAct',
'AsyncReAct', 'Sequential', 'AsyncSequential'
'Agent',
'AgentDict',
'AgentList',
'AsyncAgent',
'AgentForInternLM',
'AsyncAgentForInternLM',
'MathCoder',
'AsyncMathCoder',
'ReAct',
'AsyncReAct',
'Sequential',
'AsyncSequential',
'StreamingAgent',
'StreamingSequential',
'AsyncStreamingAgent',
'AsyncStreamingSequential',
]
Loading

0 comments on commit 44465ce

Please sign in to comment.