darsoarafa commited on
Commit
01e988c
·
verified ·
1 Parent(s): d353981

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -7
app.py CHANGED
@@ -35,6 +35,7 @@ from pydantic_ai.messages import (
35
  )
36
  from pydantic_ai.models.openai import OpenAIModel
37
 
 
38
  model = OpenAIModel(
39
  'gemma-2-2b-it', #'mistral-7b-instruct-v0.3',
40
  base_url='http://localhost:1234/v1', #LM Studio
@@ -70,7 +71,6 @@ class SalesOrder(BaseModel):
70
  ds_totalAmount:Optional[float] = 0
71
  #ds_paymentTerms: Optional[str] = ""
72
 
73
- #os.environ['GEMINI_API_KEY'] = 'AIzaSyAsVIHsPIIfDBTb2K6VNdNlMt05t8x3mtE'
74
  #agent= Agent('gemini-1.5-flash', result_type=SalesOrder)
75
 
76
  # # Create a system prompt to guide the model
@@ -83,13 +83,8 @@ Anda adalah asisten yang ahli melakukan ekstrak data SalesOrder dari teks user.
83
 
84
  #INI SAJA. SALAH SATU
85
  agent = Agent('gemini-1.5-flash', system_prompt=SYSTEM_PROMPT) # OK-Gemini
86
- #agent = Agent(model) # OK-Lokal
87
 
88
-
89
-
90
- #agent = Agent(model, result_type=SalesOrder) # belum bisa untuk local
91
- #agent = Agent(model, system_prompt=SYSTEM_PROMPT) # belum bisa untuk local
92
- #agent= Agent('gemini-1.5-flash', system_prompt=SYSTEM_PROMPT) # ERR
93
  THIS_DIR = Path(__file__).parent
94
 
95
 
 
35
  )
36
  from pydantic_ai.models.openai import OpenAIModel
37
 
38
+ # Model Local untuk LLM
39
  model = OpenAIModel(
40
  'gemma-2-2b-it', #'mistral-7b-instruct-v0.3',
41
  base_url='http://localhost:1234/v1', #LM Studio
 
71
  ds_totalAmount:Optional[float] = 0
72
  #ds_paymentTerms: Optional[str] = ""
73
 
 
74
  #agent= Agent('gemini-1.5-flash', result_type=SalesOrder)
75
 
76
  # # Create a system prompt to guide the model
 
83
 
84
  #INI SAJA. SALAH SATU
85
  agent = Agent('gemini-1.5-flash', system_prompt=SYSTEM_PROMPT) # OK-Gemini
86
+ #agent = Agent(model, system_prompt=SYSTEM_PROMPT) # OK-Lokal
87
 
 
 
 
 
 
88
  THIS_DIR = Path(__file__).parent
89
 
90