Skip to content

Commit

Permalink
fix: add correct dependencies and missing variable (#1638)
Browse files Browse the repository at this point in the history
  • Loading branch information
goetzrobin authored Aug 15, 2024
1 parent 738978a commit 15e7b64
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 1 deletion.
2 changes: 1 addition & 1 deletion memgpt/llm_api/openai.py
Original file line number Diff line number Diff line change
Expand Up @@ -252,7 +252,7 @@ def _sse_post(url: str, data: dict, headers: dict) -> Generator[ChatCompletionCh
# Inspect for errors before iterating (see https://github.com/florimondmanca/httpx-sse/pull/12)
if not event_source.response.is_success:
# handle errors
from utils import printd
from memgpt.utils import printd

printd("Caught error before iterating SSE request:", vars(event_source.response))
printd(event_source.response.read())
Expand Down
1 change: 1 addition & 0 deletions memgpt/server/rest_api/agents/message.py
Original file line number Diff line number Diff line change
Expand Up @@ -95,6 +95,7 @@ async def send_message_to_agent(
) -> Union[StreamingResponse, UserMessageResponse]:
"""Split off into a separate function so that it can be imported in the /chat/completion proxy."""

include_final_message = True
# handle the legacy mode streaming
if stream_legacy:
# NOTE: override
Expand Down

0 comments on commit 15e7b64

Please sign in to comment.