forked from Archives/langchain
bc7e56e8df
Supporting asyncio in langchain primitives allows for users to run them concurrently and creates more seamless integration with asyncio-supported frameworks (FastAPI, etc.) Summary of changes: **LLM** * Add `agenerate` and `_agenerate` * Implement in OpenAI by leveraging `client.Completions.acreate` **Chain** * Add `arun`, `acall`, `_acall` * Implement them in `LLMChain` and `LLMMathChain` for now **Agent** * Refactor and leverage async chain and llm methods * Add ability for `Tools` to contain async coroutine * Implement async SerpaPI `arun` Create demo notebook. Open questions: * Should all the async stuff go in separate classes? I've seen both patterns (keeping the same class and having async and sync methods vs. having class separation)
151 lines
4.1 KiB
Plaintext
151 lines
4.1 KiB
Plaintext
{
|
|
"cells": [
|
|
{
|
|
"cell_type": "markdown",
|
|
"id": "f6574496-b360-4ffa-9523-7fd34a590164",
|
|
"metadata": {},
|
|
"source": [
|
|
"# Async API for LLM\n",
|
|
"\n",
|
|
"LangChain provides async support for LLMs by leveraging the [asyncio](https://docs.python.org/3/library/asyncio.html) library.\n",
|
|
"\n",
|
|
"Async support is particularly useful for calling multiple LLMs concurrently, as these calls are network-bound. Currently, only `OpenAI` is supported, but async support for other LLMs is on the roadmap.\n",
|
|
"\n",
|
|
"You can use the `agenerate` method to call an OpenAI LLM asynchronously."
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 5,
|
|
"id": "5e49e96c-0f88-466d-b3d3-ea0966bdf19e",
|
|
"metadata": {
|
|
"tags": []
|
|
},
|
|
"outputs": [
|
|
{
|
|
"name": "stdout",
|
|
"output_type": "stream",
|
|
"text": [
|
|
"\n",
|
|
"\n",
|
|
"I'm doing well. How about you?\n",
|
|
"\n",
|
|
"\n",
|
|
"I'm doing well, thank you. How about you?\n",
|
|
"\n",
|
|
"\n",
|
|
"I'm doing well, thank you. How about you?\n",
|
|
"\n",
|
|
"\n",
|
|
"I'm doing well, thank you. How about you?\n",
|
|
"\n",
|
|
"I am doing quite well. How about you?\n",
|
|
"\n",
|
|
"\n",
|
|
"I'm doing well, thank you. How about you?\n",
|
|
"\n",
|
|
"\n",
|
|
"I'm doing great, thank you! How about you?\n",
|
|
"\n",
|
|
"\n",
|
|
"I'm doing well, thanks for asking. How about you?\n",
|
|
"\n",
|
|
"\n",
|
|
"I'm doing well, thank you. How about you?\n",
|
|
"\n",
|
|
"\n",
|
|
"I'm doing well, thank you. How about you?\n",
|
|
"\u001b[1mConcurrent executed in 1.93 seconds.\u001b[0m\n",
|
|
"\n",
|
|
"\n",
|
|
"I'm doing well, thank you. How about you?\n",
|
|
"\n",
|
|
"\n",
|
|
"I'm doing well, thank you. How about you?\n",
|
|
"\n",
|
|
"\n",
|
|
"I'm doing well, thank you. How about you?\n",
|
|
"\n",
|
|
"\n",
|
|
"I'm doing well, thank you. How about you?\n",
|
|
"\n",
|
|
"\n",
|
|
"I'm doing well, thank you. How about you?\n",
|
|
"\n",
|
|
"\n",
|
|
"I'm doing well, thank you. How about you?\n",
|
|
"\n",
|
|
"I'm doing well, thank you. How about you?\n",
|
|
"\n",
|
|
"\n",
|
|
"I'm doing well, thank you. How about you?\n",
|
|
"\n",
|
|
"\n",
|
|
"I'm doing well, thank you. How about you?\n",
|
|
"\n",
|
|
"\n",
|
|
"I'm doing great, thank you. How about you?\n",
|
|
"\u001b[1mSerial executed in 10.54 seconds.\u001b[0m\n"
|
|
]
|
|
}
|
|
],
|
|
"source": [
|
|
"import time\n",
|
|
"import asyncio\n",
|
|
"\n",
|
|
"from langchain.llms import OpenAI\n",
|
|
"\n",
|
|
"def generate_serially():\n",
|
|
" llm = OpenAI(temperature=0.9)\n",
|
|
" for _ in range(10):\n",
|
|
" resp = llm.generate([\"Hello, how are you?\"])\n",
|
|
" print(resp.generations[0][0].text)\n",
|
|
"\n",
|
|
"\n",
|
|
"async def async_generate(llm):\n",
|
|
" resp = await llm.agenerate([\"Hello, how are you?\"])\n",
|
|
" print(resp.generations[0][0].text)\n",
|
|
"\n",
|
|
"\n",
|
|
"async def generate_concurrently():\n",
|
|
" llm = OpenAI(temperature=0.9)\n",
|
|
" tasks = [async_generate(llm) for _ in range(10)]\n",
|
|
" await asyncio.gather(*tasks)\n",
|
|
"\n",
|
|
"\n",
|
|
"s = time.perf_counter()\n",
|
|
"# If running this outside of Jupyter, use asyncio.run(generate_concurrently())\n",
|
|
"await generate_concurrently() \n",
|
|
"elapsed = time.perf_counter() - s\n",
|
|
"print('\\033[1m' + f\"Concurrent executed in {elapsed:0.2f} seconds.\" + '\\033[0m')\n",
|
|
"\n",
|
|
"s = time.perf_counter()\n",
|
|
"generate_serially()\n",
|
|
"elapsed = time.perf_counter() - s\n",
|
|
"print('\\033[1m' + f\"Serial executed in {elapsed:0.2f} seconds.\" + '\\033[0m')"
|
|
]
|
|
}
|
|
],
|
|
"metadata": {
|
|
"kernelspec": {
|
|
"display_name": "Python 3 (ipykernel)",
|
|
"language": "python",
|
|
"name": "python3"
|
|
},
|
|
"language_info": {
|
|
"codemirror_mode": {
|
|
"name": "ipython",
|
|
"version": 3
|
|
},
|
|
"file_extension": ".py",
|
|
"mimetype": "text/x-python",
|
|
"name": "python",
|
|
"nbconvert_exporter": "python",
|
|
"pygments_lexer": "ipython3",
|
|
"version": "3.10.9"
|
|
}
|
|
},
|
|
"nbformat": 4,
|
|
"nbformat_minor": 5
|
|
}
|