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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -9
app.py CHANGED
@@ -36,17 +36,11 @@ from pydantic_ai.messages import (
36
  from pydantic_ai.models.openai import OpenAIModel
37
 
38
  model = OpenAIModel(
39
- 'gemma-2-2b-it',
40
- base_url='http://localhost:1234/v1',
41
- api_key='your-local-api-key',
42
- )
43
- model11 = OpenAIModel(
44
- 'mistral-7b-instruct-v0.3',
45
- base_url='http://localhost:1234/v1',
46
  api_key='your-local-api-key',
47
  )
48
 
49
- #DeepSeek Key: sk-694660c67c3947e4853019473f30240d https://api.deepseek.com
50
  # Model Pydantic untuk Sales Order
51
  class product1(BaseModel):
52
  ds_productCode: Optional[str] = "" #str = Field(..., min_length=1, max_length=20)
@@ -75,7 +69,8 @@ class SalesOrder(BaseModel):
75
  ds_shippingAddress: Optional[str] = ""
76
  ds_totalAmount:Optional[float] = 0
77
  #ds_paymentTerms: Optional[str] = ""
78
- os.environ['GEMINI_API_KEY'] = 'AIzaSyAsVIHsPIIfDBTb2K6VNdNlMt05t8x3mtE'
 
79
  #agent= Agent('gemini-1.5-flash', result_type=SalesOrder)
80
 
81
  # # Create a system prompt to guide the model
 
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
 
 
 
 
 
41
  api_key='your-local-api-key',
42
  )
43
 
 
44
  # Model Pydantic untuk Sales Order
45
  class product1(BaseModel):
46
  ds_productCode: Optional[str] = "" #str = Field(..., min_length=1, max_length=20)
 
69
  ds_shippingAddress: Optional[str] = ""
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