ChainForge/setup.py
ianarawjo b92c03afb2
Replace Dalai with Ollama (#209)
* Add basic Ollama support (#208)

* Remove trapFocus warning when no OpenAI key set

* Ensure Ollama is only visible in providers list if running locally. 

* Remove Dalai.

* Fix ollama support to include chat models and pass chat history correctly

* Fix bug with debounce on progress bar updates in Prompt/Chat nodes

* Rebuilt app and update package version

---------

Co-authored-by: Laurent Huberdeau <16990250+laurenthuberdeau@users.noreply.github.com>
2024-01-08 18:33:13 -05:00

48 lines
1.4 KiB
Python

from setuptools import setup, find_packages
def readme():
with open('README.md', encoding='utf-8') as f:
return f.read()
setup(
name='chainforge',
version='0.2.8.9',
packages=find_packages(),
author="Ian Arawjo",
description="A Visual Programming Environment for Prompt Engineering",
long_description=readme(),
long_description_content_type='text/markdown',
keywords='prompt engineering LLM response evaluation',
license="MIT",
url="https://github.com/ianarawjo/ChainForge/",
install_requires=[
# Package dependencies
"flask>=2.2.3",
"flask[async]",
"flask_cors",
"requests",
"urllib3==1.26.6",
"openai",
"anthropic",
"google-generativeai",
"dalaipy>=2.0.2",
"mistune>=2.0", # for LLM response markdown parsing
],
entry_points={
'console_scripts': [
'chainforge = chainforge.app:main',
],
},
classifiers=[
# Package classifiers
'Development Status :: 3 - Alpha',
'Intended Audience :: Developers',
'License :: OSI Approved :: MIT License',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.8',
'Programming Language :: Python :: 3.9',
'Programming Language :: Python :: 3.10',
],
python_requires=">=3.8",
include_package_data=True,
)