geronimo-pericoli commited on
Commit
1c8cf6c
verified
1 Parent(s): 345d59d

Update server.py

Browse files
Files changed (1) hide show
  1. server.py +8 -2
server.py CHANGED
@@ -16,8 +16,14 @@ import aiohttp # Necesario para las peticiones HTTP as铆ncronas
16
 
17
  import asyncio
18
 
 
19
 
 
 
20
 
 
 
 
21
  api_key = os.environ.get('AZURE_API_KEY')
22
  azure_endpoint = "https://pharmaia-gpt.openai.azure.com/"
23
  api_version = "2024-02-01"
@@ -40,7 +46,7 @@ embed_model = AzureOpenAIEmbedding(
40
 
41
  Settings.llm = llm
42
  Settings.embed_model = embed_model
43
-
44
 
45
 
46
  # Configuraci贸n de paths
@@ -104,7 +110,7 @@ async def list_retrievers(source: str = None) -> dict:
104
  """
105
  global count
106
  count += 1
107
- print(count)
108
  try:
109
  available = []
110
  for current_source, indexes in retrievers_metadata.items():
 
16
 
17
  import asyncio
18
 
19
+ import logging
20
 
21
+ logger = logging.getLogger(__name__)
22
+ logging.basicConfig(level=logging.INFO)
23
 
24
+
25
+
26
+ ##### AZURE OPENAI #####
27
  api_key = os.environ.get('AZURE_API_KEY')
28
  azure_endpoint = "https://pharmaia-gpt.openai.azure.com/"
29
  api_version = "2024-02-01"
 
46
 
47
  Settings.llm = llm
48
  Settings.embed_model = embed_model
49
+ ##### FIN AZURE OPENAI #####
50
 
51
 
52
  # Configuraci贸n de paths
 
110
  """
111
  global count
112
  count += 1
113
+ logger.info(f"馃攣 Count: {count}")
114
  try:
115
  available = []
116
  for current_source, indexes in retrievers_metadata.items():