OptimAI เป็นโมดูล Python อันทรงพลังที่ออกแบบมาเพื่อเพิ่มประสิทธิภาพโค้ดของคุณโดยการวิเคราะห์ประสิทธิภาพและให้คำแนะนำที่นำไปปฏิบัติได้ โดยใช้ประโยชน์จากโมเดลภาษาขนาดใหญ่ (LLM) เพื่อให้ข้อมูลเชิงลึกและคำแนะนำโดยละเอียดตามข้อมูลโปรไฟล์ที่รวบรวมระหว่างการดำเนินการโค้ดของคุณ โมดูลนี้รองรับตัวสร้างโปรไฟล์หลายประเภทจากแพ็คเกจ perfwatch
คุณสามารถติดตั้ง OptimAI โดยใช้ pip:
pip install optimizeai
หากต้องการใช้ OptimAI คุณต้องกำหนดค่าด้วยผู้ให้บริการ LLM และคีย์ API ที่คุณต้องการ ผู้ให้บริการ LLM ที่รองรับ ได้แก่ Google (รุ่น Gemini), OpenAI, Ollama, HuggingFace และ Anthropic สำหรับ Ollama คุณต้องติดตั้ง Ollama และจำเป็นต้องดาวน์โหลดอาร์ติแฟกต์ของโมเดลก่อนหน้านี้ด้วย
เลือกผู้ให้บริการ LLM :
llm = "google"
llm = "openai"
llm = "huggingface"
llm = "anthropic"
llm = "ollama"
เลือกรุ่น :
model = "gpt-4"
, model = "gemini-1.5-flash"
, model = "codegemma"
หรือรุ่นอื่นใดที่เฉพาะเจาะจงสำหรับผู้ให้บริการ LLM ที่เลือกตั้งค่าคีย์ API :
นี่คือตัวอย่างพื้นฐานที่สาธิตวิธีใช้ OptimAI เพื่อปรับฟังก์ชันให้เหมาะสม:
from optimizeai . decorators . optimize import optimize
from optimizeai . config import Config
from dotenv import load_dotenv
import time
import os
# Load environment variables
load_dotenv ()
llm = os . getenv ( "LLM" )
key = os . getenv ( "API_KEY" )
model = os . getenv ( "MODEL" )
# Configure LLM
llm_config = Config ( llm = llm , model = model , key = key )
perfwatch_params = [ "line" , "cpu" , "time" ]
# Define a test function to be optimized
@ optimize ( config = llm_config , profiler_types = perfwatch_params )
def test ():
for _ in range ( 10 ):
time . sleep ( 0.1 )
print ( "Hello World!" )
pass
if __name__ == "__main__" :
test ()
คุณสามารถตั้งค่าตัวแปรสภาพแวดล้อม ( LLM
, API_KEY
, MODEL
) ในไฟล์ .env
เพื่อความสะดวกในการใช้งาน:
LLM=google
API_KEY=your_google_api_key
MODEL=gemini-1.5-flash
เรายินดีต้อนรับการมีส่วนร่วมของ OptimAI! หากคุณมีแนวคิดเกี่ยวกับฟีเจอร์ใหม่หรือพบข้อบกพร่อง โปรดเปิดปัญหาบน GitHub หากคุณต้องการสนับสนุนโค้ด โปรดแยกพื้นที่เก็บข้อมูลและส่งคำขอดึง
git checkout -b feature-branch
)git commit -m 'Add new feature'
)git push origin feature-branch
)OptimAI ได้รับอนุญาตภายใต้ใบอนุญาต MIT ดูไฟล์ใบอนุญาตสำหรับรายละเอียดเพิ่มเติม