Commit a32b1a2

committed

fix typo and add missing check

1 parent cc5fe19 commit a32b1a2Copy full SHA for a32b1a2

File tree

-3

lines changed

-3

lines changed

Lines changed: 4 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -183,7 +183,7 @@ async def run_stream(`
`183`	`183`	`stream=True,`
`184`	`184`	`)`
`185`	`185`
`186`		`- # Forcefully Close the database session before yielding the response`
	`186`	`+ # Forcefully close the database session before yielding the response`
`187`	`187`	`# Yielding keeps the connection open while streaming the response until the end`
`188`	`188`	`# The connection closes when it returns back to the context manger in the dependencies`
`189`	`189`	`await self.searcher.db_session.close()`
`@@ -230,5 +230,7 @@ async def run_stream(`
`230`	`230`	`)`
`231`	`231`
`232`	`232`	`async for response_chunk in chat_completion_async_stream:`
`233`		`- yield Message(content=str(response_chunk.choices[0].delta.content), role="assistant")`
	`233`	`+ # first response has empty choices`
	`234`	`+ if response_chunk.choices:`
	`235`	`+ yield Message(content=str(response_chunk.choices[0].delta.content), role="assistant")`
`234`	`236`	`return`

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -201,7 +201,7 @@ async def run_stream(`
`201`	`201`	`stream=True,`
`202`	`202`	`)`
`203`	`203`
`204`		`- # Forcefully Close the database session before yielding the response`
	`204`	`+ # Forcefully close the database session before yielding the response`
`205`	`205`	`# Yielding keeps the connection open while streaming the response until the end`
`206`	`206`	`# The connection closes when it returns back to the context manger in the dependencies`
`207`	`207`	`await self.searcher.db_session.close()`

Comments

(0)