[Bug]: autogenstudio fails to start

[stefanstranger]

Describe the bug

When running autogenstudio ui from a conda environment I get the following error:

flaml.automl is not available. Please install flaml[automl] to enable AutoML functionalities.
Traceback (most recent call last):
File "C:\Users\stefstr\AppData\Local\miniconda3\envs\autogenstudio\lib\runpy.py", line 196, in run_module_as_main
return run_code(code, main_globals, None,
File "C:\Users\stefstr\AppData\Local\miniconda3\envs\autogenstudio\lib\runpy.py", line 86, in run_code
exec(code, run_globals)
File "C:\Users\stefstr\AppData\Local\miniconda3\envs\autogenstudio\Scripts\autogenstudio.exe_main.py", line 4, in
from autogenstudio.cli import run
File "C:\Users\stefstr\AppData\Local\miniconda3\envs\autogenstudio\lib\site-packages\autogenstudio_init.py", line 1, in
from .chatmanager import *
File "C:\Users\stefstr\AppData\Local\miniconda3\envs\autogenstudio\lib\site-packages\autogenstudio\chatmanager.py", line 11, in
from .workflowmanager import WorkflowManager
File "C:\Users\stefstr\AppData\Local\miniconda3\envs\autogenstudio\lib\site-packages\autogenstudio\workflowmanager.py", line 7, in
import autogen
File "C:\Users\stefstr\AppData\Local\miniconda3\envs\autogenstudio\lib\site-packages\autogen_init.py", line 9, in
from .agentchat import *
File "C:\Users\stefstr\AppData\Local\miniconda3\envs\autogenstudio\lib\site-packages\autogen\agentchat_init_.py", line 8, in
from .assistant_agent import AssistantAgent
File "C:\Users\stefstr\AppData\Local\miniconda3\envs\autogenstudio\lib\site-packages\autogen\agentchat\assistant_agent.py", line 11, in
from .conversable_agent import ConversableAgent
File "C:\Users\stefstr\AppData\Local\miniconda3\envs\autogenstudio\lib\site-packages\autogen\agentchat\conversable_agent.py", line 25, in
from ..code_utils import (
File "C:\Users\stefstr\AppData\Local\miniconda3\envs\autogenstudio\lib\site-packages\autogen\code_utils.py", line 23, in
from autogen import oai
File "C:\Users\stefstr\AppData\Local\miniconda3\envs\autogenstudio\lib\site-packages\autogen\oai_init_.py", line 8, in
from autogen.oai.client import ModelClient, OpenAIWrapper
File "C:\Users\stefstr\AppData\Local\miniconda3\envs\autogenstudio\lib\site-packages\autogen\oai\client.py", line 15, in
from flaml.automl.logger import logger_formatter
File "C:\Users\stefstr\AppData\Local\miniconda3\envs\autogenstudio\lib\site-packages\flaml\automl_init_.py", line 1, in
from flaml.automl.automl import AutoML, size
File "C:\Users\stefstr\AppData\Local\miniconda3\envs\autogenstudio\lib\site-packages\flaml\automl\automl.py", line 20, in
from flaml.automl.ml import huggingface_metric_to_mode, sklearn_metric_name_set, spark_metric_name_dict, train_estimator
File "C:\Users\stefstr\AppData\Local\miniconda3\envs\autogenstudio\lib\site-packages\flaml\automl\ml.py", line 12, in
from flaml.automl.model import BaseEstimator, TransformersEstimator
File "C:\Users\stefstr\AppData\Local\miniconda3\envs\autogenstudio\lib\site-packages\flaml\automl\model.py", line 19, in
import sklearn
ModuleNotFoundError: No module named 'sklearn'

Steps to reproduce

1. Create a new conda environment with: conda create -n autogenstudio python=3.10
2. Set environment variable: conda env config vars set AZURE_OPENAI_API_KEY=1234566
3. Activate environment: conda activate autogenstudio
4. Install autogenstudio: pip install autogenstudio
5. Start autogenstudio: autogenstudio ui

Model Used

Azure OpenAI gpt-4o

Expected Behavior

autogenui should start

Screenshots and logs

No response

Additional Information

OS: Windows 11
Python version: 3.11 and 3.10

[martinbel]

I'm finding the same issue. It seems the flaml dependency broke autogen.
This issue happens with autogen, not only autogenstudio.

[jochenvw]

+1

[ccran]

+1

[stefanstranger]

Temporary solution is to create a constraints.txt file with the following content:

flaml==2.2.0

run: conda install -c .\constraints.txt autogenstudio

[martinbel]

@stefanstranger how would this work with pip?

[stefanstranger]

Sorry that was a typo. Run pip install -c constraints.txt autogen Sent from Outlook for Android<https://aka.ms/AAb9ysg>

…

________________________________ From: Martin ***@***.***> Sent: Friday, September 20, 2024 6:34:17 PM To: microsoft/autogen ***@***.***> Cc: Mention ***@***.***>; Author ***@***.***>; Comment ***@***.***> Subject: Re: [microsoft/autogen] [Bug]: autogenstudio fails to start (Issue #3548) @stefanstranger<https://github.com/stefanstranger> how would this work with pip? — Reply to this email directly, view it on GitHub<#3548 (comment)> or unsubscribe<https://github.com/notifications/unsubscribe-auth/AAQG4SPN2VQQMREWE2VGKO3ZXRFATBFKMF2HI4TJMJ2XIZLTSSBKK5TBNR2WLJDUOJ2WLJDOMFWWLO3UNBZGKYLEL5YGC4TUNFRWS4DBNZ2F6YLDORUXM2LUPGBKK5TBNR2WLJDUOJ2WLJDOMFWWLLTXMF2GG2C7MFRXI2LWNF2HTAVFOZQWY5LFUVUXG43VMWSG4YLNMWVXI2DSMVQWIX3UPFYGLAVFOZQWY5LFVI2TQNJZGQ4DKOJXGGSG4YLNMWUWQYLTL5WGCYTFNSWHG5LCNJSWG5C7OR4XAZNMJFZXG5LFINXW23LFNZ2KM5DPOBUWG44TQKSHI6LQMWVHEZLQN5ZWS5DPOJ42K5TBNR2WLKJWHAYDCMRQGA3TDAVEOR4XAZNFNFZXG5LFUV3GC3DVMWVDENJTG4YDKNRXG44YFJDUPFYGLJLMMFRGK3FFOZQWY5LFVI2TQNJZGQ4DKOJXGGTXI4TJM5TWK4VGMNZGKYLUMU>. You are receiving this email because you were mentioned. Triage notifications on the go with GitHub Mobile for iOS<https://apps.apple.com/app/apple-store/id1477376905?ct=notification-email&mt=8&pt=524675> or Android<https://play.google.com/store/apps/details?id=com.github.android&referrer=utm_campaign%3Dnotification-email%26utm_medium%3Demail%26utm_source%3Dgithub>.