Compare commits
2 Commits
091e0b27e6
...
fe61b63a0b
Author | SHA1 | Date |
---|---|---|
Thamognya Kodi | fe61b63a0b | |
Thamognya Kodi | 5939e7804e |
|
@ -47,7 +47,7 @@ def answer(
|
|||
results: tuple[list[str], list[str]] = internet.Google(
|
||||
query, GOOGLE_SEARCH_API_KEY, GOOGLE_SEARCH_ENGINE_ID
|
||||
).google()
|
||||
context: str = str(" ".join([str(string) for string in results[0]]))
|
||||
context: str = str(" ".join([str(string) for string in results]))
|
||||
print(f"context: {context}")
|
||||
|
||||
if model.startswith("openai-"):
|
||||
|
@ -90,6 +90,6 @@ def answer(
|
|||
print(
|
||||
answer(
|
||||
query="What is the newest pokemon game?",
|
||||
model="hf-deepset/xlm-roberta-large-squad2",
|
||||
model="openai-chatgpt",
|
||||
)
|
||||
)
|
||||
|
|
|
@ -100,9 +100,7 @@ class Google:
|
|||
soup = BeautifulSoup(html, "html.parser")
|
||||
text = soup.get_text()
|
||||
normalized_text = normalizer(text)
|
||||
sentences: list[str] = sentencizer(normalized_text)
|
||||
sentence: str = str(" ".join(sentences))
|
||||
return sentence
|
||||
return normalized_text
|
||||
except Exception:
|
||||
error: str = ""
|
||||
return error
|
||||
|
@ -118,8 +116,9 @@ class Google:
|
|||
def __get_urls_contents(self: "Google") -> None:
|
||||
loop = asyncio.new_event_loop()
|
||||
asyncio.set_event_loop(loop)
|
||||
self.__content = loop.run_until_complete(self.__fetch_urls(self.__urls))
|
||||
contents = loop.run_until_complete(self.__fetch_urls(self.__urls))
|
||||
loop.close()
|
||||
self.__content = contents
|
||||
|
||||
def __filter_irrelevant_processing(self: "Google") -> None:
|
||||
with concurrent.futures.ThreadPoolExecutor(max_workers=500) as executor:
|
||||
|
@ -137,7 +136,7 @@ class Google:
|
|||
self.__get_urls_contents()
|
||||
if filter_irrelevant:
|
||||
self.__filter_irrelevant_processing()
|
||||
results: tuple[list[str], list[str]] = (self.__content[0], self.__urls) # type: ignore
|
||||
results: tuple[list[str], list[str]] = (self.__content, self.__urls) # type: ignore
|
||||
return results
|
||||
|
||||
|
||||
|
|
|
@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
|
|||
|
||||
[tool.poetry]
|
||||
name = "internet-ml"
|
||||
version = "0.1.0"
|
||||
version = "3.0.0" # fresh start from 3.0.0
|
||||
description = "Internet-ML: Allowing ML to connect to the internet"
|
||||
readme = "./.github/README.md"
|
||||
authors = ["Thamognya Kodi <contact@thamognya.com>"]
|
||||
|
|
Loading…
Reference in New Issue