requirements.txt clean up

pull/180/head
Kye 1 year ago
parent 9f20592612
commit 36a641b5f4

@ -7,7 +7,7 @@ repos:
rev: 'v0.0.255' rev: 'v0.0.255'
hooks: hooks:
- id: ruff - id: ruff
args: [--fix] args: [----unsafe-fixes]
- repo: https://github.com/nbQA-dev/nbQA - repo: https://github.com/nbQA-dev/nbQA
rev: 1.6.3 rev: 1.6.3
hooks: hooks:

@ -3,9 +3,6 @@ transformers
pandas pandas
langchain langchain
nest_asyncio nest_asyncio
pegasusx
google-generativeai
EdgeGPT
langchain-experimental langchain-experimental
playwright playwright
wget==3.2 wget==3.2
@ -24,9 +21,7 @@ soundfile
huggingface-hub huggingface-hub
google-generativeai google-generativeai
sentencepiece sentencepiece
duckduckgo-search
PyPDF2 PyPDF2
agent-protocol
accelerate accelerate
chromadb chromadb
tiktoken tiktoken
@ -54,7 +49,6 @@ openai
opencv-python opencv-python
prettytable prettytable
safetensors safetensors
streamlit
test-tube test-tube
timm timm
torchmetrics torchmetrics
@ -62,7 +56,6 @@ webdataset
marshmallow marshmallow
yapf yapf
autopep8 autopep8
dalle3
cohere cohere
torchvision torchvision
rich rich
@ -71,5 +64,4 @@ rich
mkdocs mkdocs
mkdocs-material mkdocs-material
mkdocs-glightbox mkdocs-glightbox
pre-commit
pre-commit

@ -135,9 +135,9 @@ def parse_done_token(response: str) -> bool:
class Flow: class Flow:
""" """
Flow is a chain like structure from langchain that provides the autonomy to language models Flow is the structure that provides autonomy to any llm in a reliable and effective fashion.
to generate sequential responses. The flow structure is designed to be used with any llm and provides the following features:
Features: Features:
* Interactive, AI generates, then user input * Interactive, AI generates, then user input
* Message history and performance history fed -> into context -> truncate if too long * Message history and performance history fed -> into context -> truncate if too long

@ -149,6 +149,7 @@ class SequentialWorkflow:
return {task.description: task.result for task in self.tasks} return {task.description: task.result for task in self.tasks}
def remove_task(self, task_description: str) -> None: def remove_task(self, task_description: str) -> None:
"""Remove tasks from sequential workflow"""
self.tasks = [ self.tasks = [
task for task in self.tasks if task.description != task_description task for task in self.tasks if task.description != task_description
] ]

Loading…
Cancel
Save