E2B Python와 Mistral Codestral로 AI 코드 실행하기
E2B Python와 Mistral Codestral로 AI 코드 실행하기 (E2B Python AI Code Execution with Mistral's Codestral)
Mistral의 Codestral을 LLM으로, E2B의 Code Interpreter SDK로 코드 해석 기능을 사용해 CSV 데이터 기반 선형 회귀 차트를 그리는 AI 데이터 분석가를 만드는 문서예요 (Python 버전).
출처: 문서
본문
이 AI 데이터 분석가는 CSV 데이터를 기반으로 선형 회귀 차트를 그릴 수 있어요. LLM으로는 Mistral의 Codestral을, 코드 해석 능력으로는 E2B의 [Code Interpreter SDK]를 사용해요. 이 SDK는 [Firecracker]로 구동되는 안전한 클라우드 샌드박스를 빠르게 만들며, 이 샌드박스 안에는 LLM이 사용할 수 있는 실행 중인 Jupyter 서버가 있습니다.
Mistral의 새 Codestral 모델에 대해 더 알아보려면 [여기]를 참고하세요.
AI 에이전트는 업로드된 CSV 파일에 대한 데이터 분석 작업을 수행하고, E2B 샌드박스 환경에서 AI가 생성한 코드를 실행하며, 그 결과 차트를 PNG 파일로 저장해 반환해요.
설치 (Installation)
1. API 키 로드
프로그램의 해당 부분에 API 키를 추가해요.
- [E2B API KEY] 발급
- [MISTRAL API KEY] 발급
2. 프로그램 실행
VSCode에서 Python Jupyter Notebook을 사용하려면 Jupyter 패키지가 설치된 Anaconda 환경이나 다른 Python 환경을 활성화해야 해요. Run 아이콘으로 개별 셀을 실행할 수 있으며, 출력은 코드 셀 아래에 표시됩니다.
스크립트는 다음 단계를 수행해요:
- 환경 변수에서 API 키 로드
- CSV 데이터셋을 E2B 샌드박스 클라우드 환경에 업로드
- Codestral 모델에 프롬프트를 보내 데이터셋 분석용 Python 코드 생성
- E2B Code Interpreter SDK로 생성된 Python 코드 실행
- 생성된 시각화를 PNG 파일로 저장
프로그램 실행 후에는 데이터 분석 작업의 결과가 image_1.png 파일에 저장됩니다. 이와 같은 그래프를 볼 수 있을 거예요.
E2B와 연결하고 더 알아보기
문제가 생기면 [Discord]에서 알려주세요. Code Interpreter SDK 사용법에 대해 더 알아보려면 [E2B 문서]를 확인하세요.